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日志函数: