Pytorch使用Visdom进行数据可视化的示例代码
更新时间:2023年12月12日 10:09:39 作者:思影影思
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数据可视化的资料请关注脚本之家其它相关文章!
相关文章
Python django框架输入汉字,数字,字符生成二维码实现详解
这篇文章主要介绍了Python django框架输入汉字,数字,字符转成二维码实现详解,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2019-09-09
深入理解Python中的range()函数用法、原理与实战指南
range()函数是Python中生成数字序列的基础工具,支持多种调用方式和参数组合,适用于循环控制、数据处理、性能优化和高级应用,掌握其用法和特性,可以提高代码的效率和可读性,本文给大家介绍Python中的range()函数用法、原理与实战指南,感兴趣的朋友跟随小编一起看看吧2025-12-12


最新评论