咋看训练曲线|可视化训练效果的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日志函数:

image-20210621122009363

<
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值