pytorch中实时可视化窗口Visdom的打开方式

1、安装Visdom包

首先在Anaconda命令行下激活pytorch环境，在pytorch环境下安装visdom所需要的包。如下图所示，为安装成功界面。

 

 2、如何启动

由于visdom的运行需要终端服务器的支持，所以需要在当前命令行输入python -m visdom.server打开一个终端监听窗口，并将返回的网址复制到浏览器即可。

3、新建visdom窗口，实时显示loss函数的变化

from visdom import Visdom

if __name__ == '__main__':
    wind = Visdom()
    wind.line([0.],  # Y的第一个点的坐标
              [0.],  # X的第一个点的坐标
              win="loss",  # 窗口的名称
              opts=dict(title='loss'))
    for epoch in range(1000):
        for i,data in enumerate(train_loader,0):
            inputs,labels = data
            y_pred = model(inputs)
            loss = criterion(y_pred,labels)
            print(epoch,i,loss.item())

            optimizer.zero_grad()
            loss.backward()
            optimizer.step()
        wind.line([loss.item()],[epoch],win="loss",update='append')