使用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