本地浏览器使用tensorboard查看远程服务器训练情况

前言

由于服务器端是没有浏览器的(纯命令模式),我们需要进行相应的配置,才可以在本地浏览器,使用tensorboard查看服务器运行的训练过程。

方案:Xshell隧道

配置

  • 打开Xshell,右击相应的会话,在弹出的对话框中选择属性->连接->SSH->隧道->添加。
  • 在侦听端口和目标端口中填入相同的端口,此处填写了6006;(这里的端口号,也可以随便换成其他的,只要保证两处相同即可)。
  • 本地浏览器使用tensorboard查看远程服务器训练情况

服务器端启动tensorboard

使用以下语句启动:
tensorboard --logdir=log地址 --port=6006

注意:logdir是log存放地址,要使用tensorboard,训练过程的代码中肯定有保存log的步骤,port就是之前设置的端口号

本地浏览器查看

然后再本地浏览器中输入:http://127.0.0.1:6006 或者localhost:6006,既可以通过tensorboard查看目前的训练情况。

参考:

https://blog.****.net/a314688122a/article/details/81505082