使用tensorboard(pytorch) 画拆线图

第一步：代码如下

 

from torch.utils.tensorboard import SummaryWriter

writer = SummaryWriter("logs")      # logs是创建存储图文件的文件夹名称

for i in range(10):
    writer.add_scalar("title", i*2, i)   #  参数依次是标题，y轴，x轴

writer.close()

第二步：tensorboard --logdir="logs"

打开terminal:   cd 到代码当前目录，  最后输入tensorboard --logdir="logs"，  

最后点击，localhost:6006，进入网页，即可。