python

关注公众号 jb51net

关闭
首页 > 脚本专栏 > python > Pytorch visdom可视化

Pytorch使用visdom可视化问题

作者:乐亦亦乐

这篇文章主要介绍了Pytorch使用visdom可视化问题,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教

visdom安装

pip install visdom

安装成功后,适应命令开启visdom服务,该服务基于Web,默认使用8097端口

python -m visdom.server

访问给出的地址

例子

完整代码

import torch
import visdom
 
 
#创建visdom客户端,使用默认端口8097,环境为first,环境的作用是对可视化的空间进行分区
vis=visdom.Visdom(env='first')
 
 
#vis对象有text()、line()和image()等函数,其中的win参数代表了显示的窗格(pane)的名字
vis.text('Hello visdom',win='text1')
 
 
#在此处使用append为true来增加text,否则会覆盖之前的text
vis.text('hello Pytorch',win='text1',append=True)
 
#绘制y=-i^2+20Xi+1的曲线,opts可以进行标题、坐标轴标签等的配置
for i in range(20):
         vis.line(X=torch.FloatTensor([i]),Y=torch.FloatTensor([-i**2+20*i+1]),
             opts={'title':'y=-x^2+20x+l'},win='loss',update='append')
 
 
#可视化一张随机图片
vis.image(torch.randn(3,256,256),win='random_image')

总结

以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。

您可能感兴趣的文章:
阅读全文