使用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,进入网页,即可。

 

 

上一篇:10、TORCH.UTILS.TENSORBOARD


下一篇:安装tensorboard的细节问题