Pytorch中TensorBoard及torchsummary的使用详解


Posted in Python onMay 12, 2021

1.TensorBoard神经网络可视化工具

TensorBoard是一个强大的可视化工具,在pytorch中有两种调用方法:

1.from tensorboardX import SummaryWriter

这种方法是在官方还不支持tensorboard时网上有大神写的

2.from torch.utils.tensorboard import SummaryWriter

这种方法是后来更新官方加入的

1.1 调用方法

1.1.1 创建接口SummaryWriter

功能:创建接口

调用方法:

writer = SummaryWriter("runs")

参数:

log_dir:event file输出文件夹

comment:不指定log_dir时,文件夹后缀

filename_suffix:event file文件名后缀

1.1.2 记录标量add_scalars()

功能:记录标量add_scalars()

调用方法:

writer.add_scalars("name",{"dic":val},epoch)

参数:

tag:图像的标签名

scalar_step:要记录的标量

global_step:轮次

1.1.3 统计直方图add_histogram()

功能:统计直方图与多分位数折线图

调用方法:

writer.add_histogram("weight",self.fc.weight,epoch)

参数:

tag:图像的标签名

values:要画直方图的数据

global_step:轮次

bins:取值有 ‘tensorflow'、‘auto'、‘fd' 等

1.1.4 批次显示图像add_image()

功能:批次显示图像

调用方法:

writer.add_image(“Cifar10”, img_batch, epoch,'CHW')

参数:

tag:图像的标签名

img_tensor:图像数据,注意尺寸

global_step:轮次

dataformats:数据形式,CHW,HWC,HW

1.1.5 查看模型图add_graph()

功能:查看模型图

调用方法:

writer.add_graph(model=net,input_to_model=torch.randn(1,3, 224, 224).to(device))

参数:

model:模型,必须是nn.Module

input_to_model:输出给模型的数据

verbose:是否打印计算图结构信息

写完记得要写 writer.close()

2.查看网络层形状、参数torchsummary

功能:查看网络层形状、参数

调用方法:

from torchsummary import summary
summary(net, input_size=(3, 224, 224))

参数:

model:pytorch模型

input_size:模型输入size

batch_size:batch size

device:“cuda” or “cpu”

3.启动tensorboard

在文件路径中cmd打开终端,输入

tensorboard --logdir="./runs"

runs是我保存文件的文件名,打开以下链接

Pytorch中TensorBoard及torchsummary的使用详解

补充:pytorch调用tensorboard方法尝试

tensorboard提供了用于监视训练损失很好的接口,可以帮助我们更好的调整参数。下文介绍如何在pytorch中调用tensorboard。

首先

安装tensorboard、tensorflow以及tensorboardX

第二

在文件开头导入SummaryWriter

from tensorboardX import SummaryWriter

第三

同tensorflow的tensorboard一样,tensorboardX提供多种记录方式如scalar、image等。

writer = SummaryWriter('path')

如果不添加path,则默认以时间命名。

第四

添加监视变量

writer.add_scalar('Train/Acc', Acc, iter)

第五

打开tensorboard

tensorboard --logdir 'path'

第六

在浏览器打开6006端口

Pytorch中TensorBoard及torchsummary的使用详解

以上为个人经验,希望能给大家一个参考,也希望大家多多支持三水点靠木。如有错误或未考虑完全的地方,望不吝赐教。

Python 相关文章推荐
详解Python中break语句的用法
May 14 Python
Python中title()方法的使用简介
May 20 Python
python 容器总结整理
Apr 04 Python
详解Python 实现元胞自动机中的生命游戏(Game of life)
Jan 27 Python
Python使用遗传算法解决最大流问题
Jan 29 Python
python读取图片并修改格式与大小的方法
Jul 24 Python
python实现飞机大战微信小游戏
Mar 21 Python
想学python 这5本书籍你必看!
Dec 11 Python
python将控制台输出保存至文件的方法
Jan 07 Python
关于django 1.10 CSRF验证失败的解决方法
Aug 31 Python
5分钟快速掌握Python定时任务框架的实现
Jan 26 Python
利用Python读取微信朋友圈的多种方法总结
Aug 23 Python
pytorch 一行代码查看网络参数总量的实现
May 12 #Python
pytorch查看网络参数显存占用量等操作
May 12 #Python
Python入门之使用pandas分析excel数据
May 12 #Python
将Python代码打包成.exe可执行文件的完整步骤
python3实现Dijkstra算法最短路径的实现
pytorch 中autograd.grad()函数的用法说明
python3实现无权最短路径的方法
You might like
德生1994机评
2021/03/02 无线电
PHP编程中的常见漏洞和代码实例
2014/08/06 PHP
laravel 创建命令行命令的图文教程
2019/10/23 PHP
如何判断图片地址是否失效
2007/02/02 Javascript
WEB 浏览器兼容 推荐收藏
2010/05/14 Javascript
jquery实现网站超链接和图片提示效果
2013/03/21 Javascript
JS实现超过长度限制后自动跳转下一款文本框的方法
2015/02/23 Javascript
JavaScript变量的作用域全解析
2015/08/14 Javascript
javascript加减乘除的简单实例
2016/07/12 Javascript
AngularJS基础 ng-keydown 指令简单示例
2016/08/02 Javascript
react-native fetch的具体使用方法
2017/11/01 Javascript
Node.js搭建小程序后台服务
2018/01/03 Javascript
如何利用ES6进行Promise封装总结
2019/02/11 Javascript
webpack 代码分离优化快速指北
2019/05/18 Javascript
layer.confirm()右边按钮实现href的例子
2019/09/27 Javascript
浅析vue-cli3配置webpack-bundle-analyzer插件【推荐】
2019/10/23 Javascript
使用PreloadJS加载图片资源的基础方法详解
2020/02/03 Javascript
asyncio 的 coroutine对象 与 Future对象使用指南
2016/09/11 Python
python+selenium实现163邮箱自动登陆的方法
2017/12/31 Python
python3.4实现邮件发送功能
2018/05/28 Python
python中字符串的操作方法大全
2018/06/03 Python
Python 将json序列化后的字符串转换成字典(推荐)
2020/01/06 Python
解决echarts中饼图标签重叠的问题
2020/05/16 Python
pandas将list数据拆分成行或列的实现
2020/12/13 Python
美国最大的万圣节服装网站:HalloweenCostumes.com
2017/10/12 全球购物
英国领先的名牌服装折扣零售商:Brown Bag Clothing
2019/01/08 全球购物
英国领先的维生素和补充剂品牌:Higher Nature
2019/08/26 全球购物
HolidayLettings英国:预订最好的度假公寓、别墅和自助式住宿
2019/08/27 全球购物
GAZMAN官网:澳大利亚领先的男装品牌
2019/12/19 全球购物
《飞向蓝天的恐龙》教学反思
2014/04/09 职场文书
企业员工爱岗敬业演讲稿
2014/08/26 职场文书
优秀教育工作者事迹材料
2014/12/24 职场文书
嘉宾邀请函
2015/01/31 职场文书
社区服务活动报告
2015/02/05 职场文书
初中军训感言
2015/08/01 职场文书
SQL实现LeetCode(178.分数排行)
2021/08/04 MySQL