Pytorch：visdom

visdom作为可视化工具，效率更高。

（1）install

在命令行窗口使用 pip install visdom

（2）开始进程

在程序运行前，开启visdom。使用-m visdom server

 

 然后复制给到的链接服务器打开

（3）使用

第一种方法：画一条直线。

viz=Visdom()  #创建visdom的一个实例
global_step=0.1
# 创建一条直线，前面是Y后面是X，win是标志符，可以理解为名字； 后面title是名字
viz.line([0.],[0.],'win=test_loss',opts=dict(title='test loss'))
# 把数据添加到直线中，对应的就是 Y,X   append是指定操作添加在直线后面，不然会覆盖
viz.line([loss.item()],[global_step],win='test loss',update='append')

第二种方法，画多条线，即横坐标一样，画两条线。

viz=Visdom()  #创建visdom的一个实例
global_step=0.1
viz.line([[0.0,0.0]],[0.],win='test',opts=dict(title='test loss&&acc',legend=['loss','acc']))
viz.line([[test_loss,correct/len(test_loader.dataset)]],
         [global_step],win='test',update='append')

 

第三种：可视化功能，把某些数据可视化

viz=Visdom()  #创建visdom的一个实例
global_step=0.1
viz.images(data.view(-1,1,28,28),win='x')
viz.text(str(pred.detach().cpu().numpy()),win='pred',
         opts=dict(title='pred'))