pytorch 画loss曲线_Pytorch使用tensorboardX可视化。超详细!!!

1 引言

我们都知道tensorflow框架可以使用tensorboard这一高级的可视化的工具,为了使用tensorboard这一套完美的可视化工具,未免可以将其应用到Pytorch中,用于Pytorch的可视化。

本文主要是针对该解决方案提供一些介绍。

TensorboardX支持scalar, image, figure, histogram, audio, text, graph, onnx_graph, embedding, pr_curve and videosummaries等不同的可视化展示方式,具体介绍移步至项目Github 观看详情。

2 环境安装

本教程代码环境依赖:

python 3.6+

Pytorch 0.4.0+

tensorboardX:   pip install tensorboardX、pip install tensorflow

3 代码教程

TensorboardX可以提供中很多的可视化方式,本文主要介绍scalar 和 graph,其他类型相似。

3.1 scalar教程

01_scalar_base.py 代码

对上述代码进行解释,首先导入:from tensorboardXimport SummaryWriter,然后定义一个SummaryWriter() 实例。在SummaryWriter()上鼠标ctrl+b我们可以看到SummaryWriter()的参数为:def __init__(self, log_dir=None, comment='', **kwargs): 其中log_dir为生成的文件所放的目录,comment为文件名称。默认目录为生成runs文件夹目录。我们运行上述代码:生成结果为:

当我们为SummaryWriter(comment='base_scalar')。生成结果为:

当我们为SummaryWriter(log_dir='scalar') 添加log_dir参数,可以看到第二条数据的文件名称包括了base_scalar值。生成结果目录为:

接着解释writer.add_scalar('scalar/test', np.random.rand(), epoch),这句代码的作用就是,将我们所需要的数据保存在文件里面供可视化使用。 这里是Scalar类型,所以使用writer.add_scalar(),其他的队形使用对应的函数。第一个参数可以简单理解为保存图的名称,第二个参数是可以理解为Y轴数据,第三个参数可以理解为X轴数据。当Y轴数据不止一个时,可以使用writer.add_scalars().运行代码之后生成文件之后,我们在runs同级目录下使用命令行:tensorboard --logdir runs. 当SummaryWriter(log_dir='scalar')的log_dir的参数值 存在时,将tensorboard --logdir runs 改为 tensorboard --logdir 参数值

最后调用writer.close()。

点击链接即可看到我们的最终需要的可视化结果。

可视化效果

可以分别点击对应的图片查看详情。可以看到生成的Scalar名称为'scalar/test'与'scalar/test'一致。注:可以使用左下角的文件选择你想显示的某个或者全部图片。

3.2 Graph教程

02_graph_model.py 代码

首先我们定义一个神经网络取名为Net1。然后将其添加到tensorboard可是可视化中。

with SummaryWriter(comment='Net1')as w:

w.add_graph(model, (dummy_input,))

我们重点关注最后两句话,其中使用了python的上下文管理,with 语句,可以避免因w.close未写造成的问题。推荐使用此方式。

因为这是一个神经网络架构,所以使用 w.add_graph(model, (dummy_input,)),其中第一个参数为需要保存的模型,第二个参数为输入值,元祖类型。打开tensorvboard控制台,可得到如下结果。

点击Net1部分可以将其网络展开,查看网络内部构造。

模型可视化结果

其他部分可以继续一次展开查看详情。

3.3 网络训练综合教程

解释完上述两部分知识之后,我们可以综合运用上述两部分内容,实现线性拟合的训练过程中的loss可视化和模型的保存。

03_linear_regression.py 代码

我们将运行过程中loss和model分别保存,最后打开tensorboard控制台,可以得到模型结果和loss的结果为下图.

模型可视化结果

注:不同的graph的可视化可以使用上图Run旁边的下拉框选择。

loss可视化结果

3.4 Tensorboard综合Demo

本Demo代码为TensorboardX提供的官方Demo代码。

demo.py

最终运行的所有可视化结果为:

最终可视化结果

4 结束语

本文是我写的第一篇比较完整的文章,大家如果觉得帮助到了你,帮忙点个赞呗。

若转载,请标明出处,谢谢!!

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.mzph.cn/news/270921.shtml

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

电脑技巧:电脑键盘F1~F12按键的妙用

目录 F1:帮助键 F3:搜索按键 F4:打开浏览器历史列表 F5:刷新功能 F6:定位地址栏 F7:在“命令提示符”中调用历史指令 F8:启动系统高级菜单 F9:无 F10:需要与Shift组合使用&#xff0…

linux vim基本操作,vim基本操作笔记

在Linux系统中有多种代码编辑器,例如vim, gedit, emacs。这这些编辑器各有所长,就我个人而言,对于比较短的代码,一般可以用vim解决就不用其它的工具,而长代码的情况下更喜欢用gedit,这个gnome自带的代码编辑…

iOS 获取当前对象所在的VC

id next [self nextResponder] ;while (next ! nil) {next [next nextResponder];if ([next isKindOfClass:[XX_ViewController class]]) {//return;}}转载于:https://www.cnblogs.com/mapanguan/p/5853986.html

eureka 其它语言_SpringCloud之Eureka-Go语言中文社区

一、使用方法:1、添加maven依赖org.springframework.cloudspring-cloud-starter-netflix-eureka-server版本一般交由spring-cloud-dependencies管理。注意这个依赖的artifactId在Edgware以前是spring-cloud-starter-eureka-server,而在之后变成了spring-cloud-start…

操作系统:Win10系统下LocalNow和Roaming文件夹介绍

Win10操作系统下AppData文件夹包括以下子文件夹 - 漫游,本地和本地。 几乎每个在Win10 PC上安装的程序都会在AppData文件夹中创建自己的文件夹,并将其所有相关信息存储在其中。AppData或应用程序数据是Windows 10中的一个隐藏文件夹,可帮助保…

c语言des算法实验报告,C语言实现DES算法实验报告解析.doc

C语言实现DES算法实验报告解析xx工程大学实验报告(2015-2016学年第一学期)报告题目: DES加密算法课程名称: 密码学B任课教员:专 业:学 号:姓 名:二O一六年一月十八日一、课程概述目的:培养学员的…

[noip2010]关押罪犯 并查集

第一次看的时候想到了并查集&#xff0c;但是不知道怎么实现&#xff1b; 标解&#xff0c;f[i]表示i所属的集合&#xff0c;用f[in]表示i所属集合的补集&#xff0c;实现的很巧妙&#xff0c;可以当成一个使用并查集的巧妙应用&#xff1b; 1 #include<iostream>2 #incl…

jvm什么是本地方法

一&#xff1a;什么是本地方法 二&#xff1a;举例 三&#xff1a;为什么要使用Native Method

SQLServer:用户自定义数据类型用法

今天给大家梳理一下SQLServer:用户自定义数据类型用法&#xff0c;希望对大家能有所帮助&#xff01;1、基于基本数据类型创建的别名数据类型-- 创建生日的数据类型 CREATE TYPE birthday FROM datetime NULL; -- 创建用户表 CREATE TABLE userInfo (id varchar(32), userNam…

python fsolve说明_Python fsolve()抱怨形状.为什么?

具有函数f(x,y,z),我需要解决限制f(x,y,z) 0然后绘制它.我试图为每对(y,z)找到f(x,y,z) 0的值x&#xff1a;from numpy import *from scipy.optimize import fsolvedef func(x,y,z):return xyzy linspace(0,1,100)z linspace(0,1,100)x0 zeros((y.size,z.size)) 0.5 # the …

C语言实现与功能的程序,用C语言实现Ping程序功能

2001 年 10 月 01 日大部分人用ping命令只是作为查看另一个系统的网络连接是否正常的一种简单方法。在这篇文章中&#xff0c;作者将介绍如何用C语言编写一个模拟ping命令功能的程序。ping命令是用来查看网络上另一个主机系统的网络连接是否正常的一个工具。ping命令的工作原理…

数据库知识:SQLServer变量相关知识介绍

今天给大家分享SQLServer变量相关介绍&#xff0c;希望对大家能有所帮助&#xff01;1、概述SQLServer变量对应内存中的一个存储空间。它和常量不同&#xff0c;变量的值可以在执行过程中改变。2、分类SQLServer变量根据作用范围不同主要分为局部变量和全局变量。2.1.局部变量局…

oauth基本流程和原理

组装loginurl-》去第三方登录验证-》回调callbackurlcode&#xff08;票据&#xff09;-》本地根据codeappidappkey组装url隐式curl获取用户信息-》完成登录。 之所以要用callbackurl的形式是因为不能让用户在你本地输入账号密码防止被盗号。 之所以不在callbackurl的时候&…

halcon轮廓擦除_halcon第十二讲,毛刺去除

在视觉测量中我们很会被毛刺问题困扰&#xff0c;产生干扰的因素很多&#xff0c;有打光问题和产品本生带毛刺&#xff0c;接下来给兄弟们带来两种去毛刺的方法。第一种read_image (Image1323153Ba0b96cf52f88f1, 1)rgb1_to_gray (Image1323153Ba0b96cf52f88f1, GrayImage)lapl…

c语言双循环计算n的阶乘,用C语言用循环实现N的阶乘

匿名用户1级2014-03-04 回答不要用数组&#xff0c;用vector来实现相应的功能&#xff0c;我给你写个大概的框架&#xff0c;你向里面添加就好了1234567891011121314151617181920212223242526272829303132333435363738394041424344454647#include #include #include …

存储知识:U盘的实际容量要比标注容量小,你知道原因吗?

用过U盘的朋友一定都会知道&#xff0c;U盘的实际容量往往是达不到它标注的容量。比如一个16G的U盘&#xff0c;其实际容量一般只有14.7G左右。今天本文会给大家解释其中的原因。 原因一&#xff1a;计算容量方法不同 存储设备制造商为了计算方便&#xff0c;采用十进制计算方法…

CentOS 7 防止端口自动关闭

tl;dr firewall-cmd --permanent --zonepublic --add-port2888/tcp firewall-cmd --reload #重新载入服务 永久配置firewalld开启端口。 之前的一些坑 之前的一篇文章CentOS 7部署Node.jsMongoDB&#xff1a;在VPS上从安装到Hello world中&#xff0c;讲到了CentOS开启端口用这…

jio文件 久其_功能强大的Windows文件管理器工具

Windows文件管理功能体验只能说中规中矩&#xff0c;批量处理文件时非常的慢&#xff0c;而且在模糊搜索文件时的速度以及资源占用实在是感人&#xff0c;再加上功能较少&#xff0c;只能实现重命名、复制、粘贴、剪切这些基础功能&#xff0c;因此想要提高文件之间处理效果通常…