Pytorch使用tensorboardX实现loss曲线可视化。超详细!!!

https://www.jianshu.com/p/46eb3004beca
使用到的代码:
writer=SummaryWriter()
writer.add_scalar('scalar/test',loss,epoch) ###tensorboardX #第一个参数可以简单理解为保存图的名称,第二个参数是可以理解为Y轴数据,第三个参数可以理解为X轴数据。
#当Y轴数据不止一个时,可以使用writer.add_scalars()
writer.close() ###tensorboardX

找到文件的路径后,

在命令行(在终端的那个环境下哦 conda activate pytorch_envs)输入以下内容,能得到一个链接。
tensorboard --logdir=D:\GLP\9classify\runs\Aug21_15-57-22_DESKTOP-RQFPBN6

用浏览器打开链接就行

posted @ 2021-08-21 15:57  ice猫猫3  阅读(1288)  评论(0编辑  收藏  举报