tensorboard使用

from torch.utils.tensorboard import SummaryWriter

writer = SummaryWriter('logs')

for i in range(100):
    writer.add_scalar('y=x',i,i)

writer.close()

官方帮助文档：

相当于Add scalar data to summary.

Args:
    tag (string): Data identifier  # 生成图像的标题
    scalar_value (float or string/blobname): Value to save   # 标量值，相当于Y轴
    global_step (int): Global step value to record   #相当于横坐标

运行代码后：

--logdir 参数传入event files 所在的文件夹

点击蓝色超链接即可看到结果