1、Pytorch中使用Tensorboard
from torch.utils.tensorboard import SummaryWriter
writer =SummaryWriter("logs")
for i in range(1000):
writer.add_scalar("y=x",i,i) #这里在实际应用中是写入loss和精度
writer.close()
cmd中输入
tensorboard --logdir=logs
2、在本地浏览器打开远程服务器上开启的tensorboard服务。
xshell链接远程服务器,属性—>SSH—>隧道—>添加
测试结果。