Pytorch中使用Tensorboard,并且在本地打开远程服务器上的tensorboard

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—>隧道—>添加
Pytorch中使用Tensorboard,并且在本地打开远程服务器上的tensorboard_第1张图片

测试结果。

你可能感兴趣的:(深度学习,Pytorch,tensorboard)