pytorch使用TensorboardX记录loss

简介

tensorboardX是tensorboard的一个可视化模块， TensorboardX支持scalar, image, figure, histogram, audio, text, graph, onnx_graph, embedding, pr_curve and videosummaries等不同的可视化展示方式，具体介绍移步至项目Github 观看详情。

环境安装

pip install tensorboardX

代码教程

#step1:倒入SummaryWriter
from tensorboardX import SummaryWriter
import numpy as np
#step：2初始化一个实例writer
writer = SummaryWriter('logs')#参数为指定存储路径

for i in range(100):
    #step:3记录loss
    #当数据只需要存储一个时用add_scalar()
    writer.add_scalar("test/sin",np.sin(i),i)#将我们所需要的数据保存在文件里面供可视化使用
    #当数据不止一个时使用add_scalars()
    writer.add_scalars("test1",{"sin":np.sin(i),"cos":np.cos(i)},i)
#step4：close
writer.close()

可视化

tensorboard --logdor "logs"

可能出现的错误

1、确保tensorboardx环境安装正确，在使用的过程中因为环境问题导致出错了，挑了很长时间，用conda换了个环境就好了
2、在显示的过程中一定要确保路径正确，并且用""指定路径，路径不正确会导致一下错误。

参考

https://tensorboardx.readthedocs.io/en/latest/tensorboard.html