栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

咋看训练曲线|可视化训练效果的tensorboardx

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

咋看训练曲线|可视化训练效果的tensorboardx

pytorch可用啊大家,冲!老师问你模型训练咋样了,还在把log文件看输出数字吗!快来学学可视化技巧,让同门称卷,让师弟师妹夸强,让自己获得花里胡哨的快乐debug!

mmpose里有可视化训练开关嘿嘿!一整套!


本地环境

  • ubuntu 16.04
  • conda
  • python 3.7
1 安装

安装tensoboardx和依赖

pip install tensorboardx
pip install future tensorboard

# 如果不想使用阉割版,可以加多一条
# 下载cpu版本的tensorflow
pip install tensorflow-cpu  -i https://pypi.tuna.tsinghua.edu.cn/simple  # 清华源
2 启动

在mmpose的configs中启用tensorboard日志函数:

启动tensorboardx:

# 在存放训练log文件的地方打开
tensorboard --logdir /home/yiming/mmpose/work_dirs/higherhrnet_w32_coco_512x512_udp/

点击链接即可打开

问题 启动问题

如果还有问题就重启一下:

sudo reboot
报错

只安装了tensorboardx后,运行demo报错:

这个好像是堵车了,我把原来的文件删除后,等一会再运行就没了。

安装了tensorboard增加依赖后报错:

再试一次就好。

刷新问题

可以设置,默认30s刷新一次

可视化界面

可视化界面解读,对不起,我解读的话其实还是看命令行打印的loss,肉眼扫参。

标量和指标 SCALARS

用来返回日志中的参数


后记

为了这个工具,我装了3天,终于给我装逼成功了!

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/961250.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号