方法-TensorBoard如何在PC端远程可视化服务器的训练结果

本专栏为深度学习的一些技巧,方法和实验测试,偏向于实际应用,后续不断更新,感兴趣童鞋可关,方便后续推送

简介

TensorBoard 是 TensorFlow 的可视化工具，可以帮助用户实时监控深度神经网络的训练过程,基于PyTorch的训练过程也可以可视化。假设我在服务器上进行模型的训练,并记录下summary,此时需要进行可视化相关的曲线或者plot图.如果服务器没有可视化外设,那么就只能在本机进行远程可视化了,具体方法如下.

方法

1. 服务器虚拟环境内开启tensorboard

  tensorboard --logdir {logs_dir} --port=xxxx

2. 本地运行命令，把远程服务器上端口映射到本地端口

# 8099:127.0.0.1 是本地端口， 8100是远程服务器端口
# 通常省略 -p remote_ssh_port，如果ssh端口需要指明，则需要加上
ssh user@remote_ssh_ip -L yyyy:127.0.0.1:xxxx

3. 本地浏览器地址栏输入网址,即可打开网页

http://localhost:yyyy