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 正则式使用心得
May 07 Python
python实现进程间通信简单实例
Jul 23 Python
python使用cStringIO实现临时内存文件访问的方法
Mar 26 Python
Python中用altzone()方法处理时区的教程
May 22 Python
Python实现查看系统启动项功能示例
May 10 Python
python:print格式化输出到文件的实例
May 14 Python
Python/ArcPy遍历指定目录中的MDB文件方法
Oct 27 Python
Python中的正则表达式与JSON数据交换格式
Jul 03 Python
python实现加密的方式总结
Jan 19 Python
Python 如何创建一个线程池
Jul 28 Python
Python xmltodict模块安装及代码实例
Oct 05 Python
利用Python脚本写端口扫描器socket,python-nmap
Jul 23 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
mysql中存储过程、函数的一些问题
2007/02/14 PHP
php数组总结篇(一)
2008/09/30 PHP
PHP 字符截取 解决中文的截取问题,不用mb系列
2009/09/29 PHP
PHPMailer 中文使用说明小结
2010/01/22 PHP
为PHP5.4开启Zend OPCode缓存
2014/12/26 PHP
php中 $$str 中 "$$" 的详解
2015/07/06 PHP
PHP中substr函数字符串截取用法分析
2016/01/07 PHP
phpStudy中升级MySQL版本到5.7.17的方法步骤
2017/08/03 PHP
PHP观察者模式示例【Laravel框架中有用到】
2018/06/15 PHP
通过下拉框的值来确定输入框是否可以为空的代码
2011/10/18 Javascript
input输入框的自动匹配(原生代码)
2013/03/19 Javascript
jquery基础教程之数组使用详解
2014/03/10 Javascript
JQuery记住用户名和密码的具体实现
2014/04/04 Javascript
JS实现仿京东淘宝竖排二级导航
2014/12/08 Javascript
js实现jquery的offset()方法实例
2015/01/10 Javascript
js带前后翻页的图片切换效果代码分享
2015/09/08 Javascript
使用JavaScript脚本判断页面是否在微信中被打开
2016/03/06 Javascript
Easyui Tree获取当前选择节点的所有顶级父节点
2017/02/14 Javascript
基于Vuejs和Element的注册插件的编写方法
2017/07/03 Javascript
Angular中封装fancyBox(图片预览)遇到问题小结
2017/09/01 Javascript
KOA+egg.js集成kafka消息队列的示例
2018/11/09 Javascript
基于canvas实现手写签名(vue)
2020/05/21 Javascript
[03:22]DSPL第一期精彩集锦:酷炫到底!
2014/11/07 DOTA
python实现定制交互式命令行的方法
2014/07/03 Python
Python程序员面试题 你必须提前准备!(答案及解析)
2018/01/23 Python
python和opencv实现抠图
2018/07/18 Python
python__name__原理及用法详解
2019/11/02 Python
Python异常处理机制结构实例解析
2020/07/23 Python
python实现ping命令小程序
2020/12/28 Python
加拿大大码女装购物网站:Penningtons
2020/12/26 全球购物
采用怎样的方法保证数据的完整性
2013/12/02 面试题
领导班子四风对照检查材料
2014/09/23 职场文书
详解Go语言Slice作为函数参数的使用
2021/07/02 Golang
Redis 彻底禁用RDB持久化操作
2021/07/09 Redis
Python OpenCV超详细讲解读取图像视频和网络摄像头
2022/04/02 Python
Elasticsearch 配置详解
2022/04/19 Java/Android