1. 安装
tensorboardX的安装非常简单,直接一行命令搞定:
pip install tensorboardX
2. 使用
由于模型训练通常在远程服务器上进行,而服务器上没有可视化界面,无法直接查看tensorboard的结果,因此需要采用一些措施将服务器上的界面转发到本地,这里采用ssh端口转发的方式实现。
首先借助MobaXterm实现ssh端口转发:
- 选择状态栏中的Tunneling,进入隧道设置界面,选择新建隧道:
- 配置隧道端口并保存:
上图中,Remote server/Remote port依次填入localhost和6006(和使用tensorboard命令生成记录时的端口保持一致);SSH server/SSH login/SSH port分别填入远程服务器的IP/用户名/端口号;Forward port填入本地端口号6006(也可是其他可用端口,用于接收远程端口转发)
以上就完成了ssh隧道的建立,将远程6006端口的内容转发到本地6006端口。
然后在服务器上生成tensorboard记录:在tensorboard日志目录外执行命令:
tensorboard --logdir=${YOUR TFBORAD DIR} --port=6006
最后在本地打开https://localhost:6006
即可查看tensorboard的结果。