下方是使用的代码,注意文件名不要取成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