【软件安装】pytorch使用可视化工具 visdom 方法

在 pytorch 中使用 visdom 对可视化网络训练过程十分方便,这里简要介绍下安装方法

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

你可能感兴趣的:(【软件安装】pytorch使用可视化工具 visdom 方法)