pytorch中tensorboard使用方法

from torch.utils.tensorboard import SummaryWriter

writer = SummaryWriter("logs")
# y= 2x
for i in range(100):
    writer.add_scalar("y= 2x", 2*i, i)

writer.close()

在终端中写入tensorboard --logdir=logs,点开6006网址即可
或者自定义6007
tensorboard --logdir=logs --port=6007