在 pytorch 中使用 visdom 对可视化网络训练过程十分方便,这里简要介绍下安装方法
- pip install visdom
- 在训练网络前,首先执行
python -m visdom.server
命令,执行后点进相应连接即可; - 如果是服务器,需要在本地进行端口配置,具体方法是在本地终端执行
ssh -L 18097:127.0.0.1:8097 username@host ip
,然后在本地浏览器中输入127.0.0.1:18907
即可访问,其中 8097 是服务器端的端口; - 如果还连接服务器过程还存在跳板机,只需要先绑定跳板机,然后再在跳板机上绑定服务端口号即可;
- 在代码中
import visdom
即可使用。