tensorboard包使用

tensorboard是用于深度学习的工具包，可被集成于pytorch环境中

下方是使用的代码，注意文件名不要取成tensorboard.py，否则Python会报错，文件名冲突

from torch.utils.tensorboard import SummaryWriter

writer = SummaryWriter("logs")

# writer.add_image()
# writer.add_scalar()
# y=x
for i in range(100):
    writer.add_scalar("y=x",i,i)

writer.close()

下方是在6007端口启动画图服务

tensorboard --logdir=logs -- port=6007