Tensorboard 在本机可以方便使用,但使用服务器时需要设置一下。
原理
建立ssh隧道,实现远程端口到本地端口的转发 具体来说就是将远程服务器的6006端口(tensorboard默认将数据放在6006端口)转发到本地的16006端口,在本地对16006端口的访问即是对远程6006端口的访问,当然,转发到本地某一端口不是限定的,可自由选择。
Mac或Linux系统
在登录远程服务器的时候使用命令:
ssh -L 16006:127.0.0.1:6006 account@server.address
代替一般ssh远程登录命令:ssh account@server.address。ip地址:127.0.0.1指代本地主机。
训练完模型之后使用如下命令:
tensorboard --logdir="/path/to/log-directory" --port=6006
其中,/path/to/log-directory
为自己设定的日志存放路径,因人而异
最后,在本地浏览器访问地址:http://127.0.0.1:16006/
参考博客:
https://blog.csdn.net/aiynmimi/article/details/88531695