tensorboard 查看训练过程（远程服务器）

代码部分

from torch.utils.tensorboard import SummaryWriter
writer = SummaryWriter(logs_save_dir)
# 添加训练相关参数
writer.add_scalar('train/loss', loss, global_step)
writer.add_scalar('train/acc', train_acc, global_step)
writer.add_scalar('train/lr', lr, global_step)
# 添加测试相关参数
writer.add_scalar('test/acc', test_acc, global_step)
writer.add_scalar('test/loss', test_loss, global_step)

还可以添加许多其他的信息：

远程连接 tensorboard

服务器

tensorboard --logdir [logs_save_dir]

出现以下表明成功

本地

ssh -L 16006:127.0.0.1:6006 username@remote_server_ip

建立ssh隧道，实现远程端口到本地端口的转发
具体来说就是将远程服务器的6006端口（tensorboard默认将数据放在6006端口）转发到本地的16006端口，在本地对16006端口的访问即是对远程6006端口的访问，当然，转发到本地某一端口不是限定的，可自由选择。

如果登录服务器的端口不是默认端口22，则需要加上参数 -p

ssh -p [PORT] -L 16006:127.0.0.1:6006 username@remote_server_ip

浏览器输入 http://localhost:16006/ 即可看到