其他分享
首页 > 其他分享> > 使用tensorboard(pytorch) 画拆线图

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

 

 

标签:logs,SummaryWriter,writer,拆线,--,pytorch,logdir,tensorboard
来源: https://blog.csdn.net/julicliy/article/details/117827696