visdom是由Facebook开发的PyTorch训练过程可视化工具,类似于tensorboard。本文主要记录在远程服务器上跑PyTorch模型时如何在本地访问visdom的结果。
- ssh远程服务器时绑定本地端口与远程服务器端口
# ssh -L <local port>:localhost:<remote port> -p <port> username@remote_host
ssh -L 6606:localhost:8097 -p <port> username@remote_host
- 这里的
6606
是本地的端口 -
8097
是绑定的远程服务器的端口,选用8097
的原因是visdom默认的端口是8097
,可以修改。
- 在remote host开一个tmux窗口上启动visdom服务(这一步很关键),不启动visdom服务的话会报
ConnectionError
> visdom -p <remote port> --hostname <hostname>
- 模型代码中visdom的配置
# server = <hostname>
# remote_port = <remote port>
visdom.Visdom(server=server, port = remote_port )
- 这里的
server
启动visdom
时指定的hostname
- 这里的
port
是启动visdom
时指定的remote port
(上面的8097
)
- 在local host的浏览器访问127.0.0.1:6606既可以查看模型的训练情况
N.B
- 一定要先启动
visdom
服务 - 在本地可以打开127.0.0.1:6606证明已经连接成功了,没有内容可能是第一个iter还没跑完
网友评论