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
用浏览器打开链接就行