Pytorch使用Visdom进行数据可视化的示例代码
作者：思影影思
pytorch Visdom可视化,是一个灵活的工具,用于创建,组织和共享实时丰富数据的可视化,这个博客简要介绍一下在使用Pytorch进行数据可视化的一些内容,感兴趣的朋友可以参考下
安装
pip install visdom
启动服务
python -m visdom.server
使用
基本上是按照先生成对象，然后追加内容的方式。
import visdom
vis = visdom.Visdom()
vis.line([0.], [0.], win='jax train-loss', name="train loss", opts=dict(title='jax train loss'))
vis.line([0.0], [0.], win='jax time-consumed', name="time", opts=dict(title='jax time'))
vis.text(f"jax 进行代理模型训练", win="jax log", opts={"title": "jax log"})
# jit_train_step = train_step
start_time = time.time()
s1=start_time
for epoch in range(iterations):
vis.text(f"{epoch+1}, Loss: {loss}, Time: {duration}", win="jax log", append=True)
vis.line([loss.item()*1000], [epoch+1], win="jax train-loss", update='append', name="train loss", opts={"title": "jax train loss"})
vis.line([duration], [epoch+1], win='jax time-consumed', update='append', name="time", opts={"title": 'jax time'})
下图中，则是同一个图中同时绘制两个曲线
下图演示绘制曲线
呈现效果
以上就是Pytorch使用Visdom进行数据可视化的示例代码的详细内容，更多关于Pytorch Visdom数据可视化的资料请关注脚本之家其它相关文章！