pytorch 一行代码查看网络参数总量的实现


Posted in Python onMay 12, 2021

大家还是直接看代码吧~

netG = Generator()
print('# generator parameters:', sum(param.numel() for param in netG.parameters()))
netD = Discriminator()
print('# discriminator parameters:', sum(param.numel() for param in netD.parameters()))

补充:PyTorch查看网络模型的参数量PARAMS和FLOPS等

在PyTorch中,可以使用torchstat这个库来查看网络模型的一些信息,包括总的参数量params、MAdd、显卡内存占用量和FLOPs等。

示例代码如下:

from torchstat import stat
from torchvision.models import resnet50, resnet101, resnet152, resnext101_32x8d
model = resnet50()
stat(model, (3, 224, 224))

打印信息如下:

pytorch 一行代码查看网络参数总量的实现

pytorch 一行代码查看网络参数总量的实现

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

Python 相关文章推荐
python网络编程学习笔记(六):Web客户端访问
Jun 09 Python
Python编程之字符串模板(Template)用法实例分析
Jul 22 Python
一篇文章搞懂Python的类与对象名称空间
Dec 10 Python
在Python 不同级目录之间模块的调用方法
Jan 19 Python
Python使用tkinter模块实现推箱子游戏
Oct 08 Python
使用 Python 在京东上抢口罩的思路详解
Feb 27 Python
Python 改变数组类型为uint8的实现
Apr 09 Python
python argparse传入布尔参数false不生效的解决
Apr 20 Python
Python xlwt模块使用代码实例
Jun 10 Python
Python爬虫headers处理及网络超时问题解决方案
Jun 19 Python
python 如何引入协程和原理分析
Nov 30 Python
Python LMDB库的使用示例
Feb 14 Python
pytorch查看网络参数显存占用量等操作
May 12 #Python
Python入门之使用pandas分析excel数据
May 12 #Python
将Python代码打包成.exe可执行文件的完整步骤
python3实现Dijkstra算法最短路径的实现
pytorch 中autograd.grad()函数的用法说明
python3实现无权最短路径的方法
Python入门之基础语法详解
May 11 #Python
You might like
php中加密解密DES类的简单使用方法示例
2020/03/26 PHP
[原创]静态页面也可以实现预览 列表不同的显示方式
2006/10/14 Javascript
JQuery 获得绝对,相对位置的坐标方法
2010/02/09 Javascript
js 无提示关闭浏览器页面的代码
2010/03/09 Javascript
javascript showModalDialog,open取得父窗口的方法
2010/03/10 Javascript
jQuery EasyUI NumberBox(数字框)的用法
2010/07/08 Javascript
基于jquery的划词搜索实现(备忘)
2010/09/14 Javascript
jquery简单的拖动效果实现原理及示例
2013/07/26 Javascript
JQuery记住用户名和密码的具体实现
2014/04/04 Javascript
JS 获取鼠标左右键的键值方法
2014/10/11 Javascript
javascript无刷新评论实现方法
2015/05/13 Javascript
JavaScript中数据结构与算法(五):经典KMP算法
2015/06/19 Javascript
基于jQuery实现二级下拉菜单效果
2016/02/01 Javascript
JS实现Select的option上下移动的方法
2016/03/01 Javascript
深入理解AngularJS中的ng-bind-html指令
2017/03/27 Javascript
详解nodejs微信公众号开发——2.自动回复
2017/04/10 NodeJs
angular select 默认值设置方法
2017/06/23 Javascript
BootStrap selectpicker后台动态绑定数据的方法
2017/07/28 Javascript
javascript+css3开发打气球小游戏完整代码
2017/11/28 Javascript
Node.js命令行/批处理中如何更改Linux用户密码浅析
2018/07/22 Javascript
解决Layui选择全部,换页checkbox复选框重新勾选的问题方法
2018/08/14 Javascript
详解关于Angular4 ng-zorro使用过程中遇到的问题
2018/12/05 Javascript
javascript判断一个变量是数组还是对象
2019/04/10 Javascript
vue数据初始化initState的实例详解
2019/04/11 Javascript
聊聊Vue 中 title 的动态修改问题
2019/06/11 Javascript
JS代码屏蔽F12,右键,粘贴,复制,剪切,选中,操作实例
2019/09/17 Javascript
详解Python的Flask框架中生成SECRET_KEY密钥的方法
2016/06/07 Python
Python使用pyh生成HTML文档的方法示例
2018/03/10 Python
Django上线部署之IIS的配置方法
2019/08/22 Python
多视角3D可旋转的HTML5 Logo动画
2016/03/02 HTML / CSS
营销专业应届生求职信
2013/11/26 职场文书
竞选学习委员演讲稿
2014/04/28 职场文书
导游词之山西祁县乔家大院
2019/10/14 职场文书
redis限流的实际应用
2021/04/24 Redis
springboot集成flyway自动创表的详细配置
2021/06/26 Java/Android
Mysql中有关Datetime和Timestamp的使用总结
2021/12/06 MySQL