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

Logo

汇聚全球AI编程工具,助力开发者即刻编程。

更多推荐