pytorch visdom可视化loss曲线和准确率曲线

1. 首先安装visdom库

pip install visdom

2. 安装完成终端输入命令

python -m visdom.server

此时可能长时间出现以下提示：

 解决方法：在网上下载static目录替换原先的static，然后替换找到安装的visdom库目录，目录下有个server.py文件，打开文件注释掉

 def download_scripts_and_run(): 下的download_scripts() 语句。

成功运行后如下：

 打开网址即可看到visdom可视化的界面，此时没有任何内容：

 3. 可视化loss曲线和准确率。

 首先导入visdom库：

from visdom import Visdom

 实例化：

viz = Visdom()

实例化中可以设置参数env，创建并指定大窗口显示如：viz = Visdom(env='my_wid')。

创建并初始化显示小窗口：

    viz.line([0.],[0.],win='train_loss',opts=dict(title='train loss'))
    viz.line([0.],[0.],win='acc',opts=dict(title='acc'))

之后在训练过程中step或epoch下更新线条即可：

viz.line([loss.item()],[global_step],win='train_loss',update='append')
viz.line([test_acc],[epoch],win='acc',update='append')