当前位置 博文首页 > 不太冷的莱昂的博客:在本地浏览器查看远程服务器训练时的tensor

    不太冷的莱昂的博客:在本地浏览器查看远程服务器训练时的tensor

    作者:[db:作者] 时间:2021-09-02 16:28

    看了一些博客,总结了一种最简单,不需要额外软件和配置的方法:将服务器的端口转发到我们的本地机子上来

    简单来说就是以下三步:

    首先你的训练代码中得有关于tensorboard写入日志的操作,我这里以tensorflow为例

    然后在训练时写入回调即可,这里的log_dir等会儿就是tensorboard监控的文件夹,也就是说训练保存的各种参数都被写入了这个文件夹了。

    第一步:在远程服务器终端输入:tensorboard -- logdir=log_dir --port 6006,这里的log_dir就是上面我说的日志文件夹,6006是服务器上的一个端口,如果被占用了换成别的比如6009.

    第二步:在本地机器命令终端上输入:ssh -L 16006:127.0.0.1:6006 服务器登录名@服务器IP,这里的6006就是上面你设置的服务器端口。

    第三步:在本地机浏览器窗口输入:http://127.0.0.1:16006/即可访问tensorboard

    最关键的一点是在服务器端输入tensorboard --log_dir='?' --port 6009时要先cd 到日志目录的上一级文件夹,再输入上述命令,否则打开tensorboard不显示内容

    新发现的问题,在本地终端输入ssh -L 16006:127.0.0.1:6006 服务器用户名@服务区IP后,这个终端不能关闭,关闭了tensorboard的内容刷新了就没了,保持打开状态则可以一直刷新,当然如果不小心关闭了重新在本地输入上述命令即可。

    cs