visdom 使用教程

import visdom


x = torch.arange(1, 30, 0.01)
y = torch.sin(x)
vis = visdom.Visdom(env=u'test', use_incoming_socket=False)

vis.line(y, x, win='sinx', opts={'title': 'y=sin(x)'})


#  动态绘制  关键是设置update 除了第一次绘制 之后都使用update
for i in range(10):
    x = torch.tensor([i])
    y = x**2
    vis.line(y, x, win='polynomial', update='append' if i>0 else None)
    #  time.sleep(0.5)

'''
image 绘制
二维张量 H W 或者三维张量 3 H W 分别对应黑白彩色图像
四维张量 N C H W 把多张图片拼接在一起
'''

vis.image(torch.randn(64, 64).numpy())
vis.image(torch.randn(3, 64, 64).numpy(), win='colorful')
vis.images(torch.randn(36, 3, 64, 64), nrow=6, win='random', opts={'title':'images'})

vis.text(u'使用html标签',win='visdom')

服务器端运行：

Python -m visdom.server