12-07 周四 Pytorch 使用Visdom 进行可视化

简介

在完成了龙良曲的Pytroch视频课程之后,楼主对于pytroch有了进一步的理解,比如,比之前更加深刻的了解了BP神经网络的反向传播算法,梯度、损失、优化器这些名词更加熟悉。这个博客简要介绍一下在使用Pytorch进行数据可视化的一些内容。

安装

bash 复制代码
pip install visdom

启动服务

bash 复制代码
python -m visdom.server

使用

基本上是按照先生成对象,然后追加内容的方式。

python 复制代码
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'})

下图中,则是同一个图中同时绘制两个曲线

下图演示绘制曲线

呈现效果

相关推荐
学slam的小范1 分钟前
ROS跑ORB-SLAM3遇见的问题总结
人工智能·机器人·自动驾驶
coding消烦员37 分钟前
新版 vscode 去除快捷键 Ctrl+I 显示 Copilot 的 AI 对话框
人工智能·vscode·copilot
周杰伦_Jay1 小时前
【自动驾驶开源仿真平台】Carla、AirSim、Udacity self-driving-car-sim、Apollo、Autoware。
人工智能·机器学习·自动驾驶
牛奶还是纯的好1 小时前
双目测距实战5-立体矫正
人工智能·3d
汤姆yu2 小时前
基于python的化妆品销售分析系统
开发语言·python·化妆品销售分析
无风听海2 小时前
神经网络之窗口大小对词语义向量的影响
人工智能·深度学习·神经网络
sali-tec2 小时前
C# 基于halcon的视觉工作流-章52-生成标定板
开发语言·图像处理·人工智能·算法·计算机视觉
IT古董2 小时前
【第五章:计算机视觉-项目实战之推荐/广告系统】2.粗排算法-(4)粗排算法模型多目标算法(Multi Task Learning)及目标融合
人工智能·算法·1024程序员节
newxtc2 小时前
【江苏政务服务网-注册_登录安全分析报告】
人工智能·安全·yolo·政务·1024程序员节·安全爆破