tensorboard的使用

1.安装tensorboard

pip3 install tensorboard

2.简单使用方式

from torch.utils.tensorboard import SummaryWriter

writer = SummaryWriter('logdir') #将生成的文件保存在当前路径的logdir目录下

for i in range(100):
    writer.add_scalar("y=x", i,i) #这里生成一个y=x的函数

3.在终端执行

tensorboard --logdir='/Users/pytorch/logdir'

注：可以tensorboard --logdir='/Users/pytorch/logdir' --port=6007设置在6007端口打开

4.点击上面的链接，可以打开tensorboard查看