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设计模式编程的代理模式与抽象工厂模式
Jan 16 Python
Python基于回溯法子集树模板解决最佳作业调度问题示例
Sep 08 Python
Pandas:Series和DataFrame删除指定轴上数据的方法
Nov 10 Python
python实现简易动态时钟
Nov 19 Python
python 移动图片到另外一个文件夹的实例
Jan 10 Python
python实现远程控制电脑
May 23 Python
python如何制作英文字典
Jun 25 Python
pandas的qcut()方法详解
Jul 06 Python
Python浮点数四舍五入问题的分析与解决方法
Nov 19 Python
python pyenv多版本管理工具的使用
Dec 23 Python
PyQt5如何将.ui文件转换为.py文件的实例代码
May 26 Python
Django-simple-captcha验证码包使用方法详解
Nov 28 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和ACCESS写聊天室(六)
2006/10/09 PHP
PHP删除数组中特定元素的两种方法
2019/02/28 PHP
在JS中最常看到切最容易迷惑的语法(转)
2010/10/29 Javascript
网页实时显示服务器时间和javscript自运行时钟
2014/06/09 Javascript
js图片滚动效果时间可随意设定当鼠标移上去时停止
2014/06/26 Javascript
NodeJS学习笔记之网络编程
2014/08/03 NodeJs
影响jQuery使用的14个方面
2014/09/01 Javascript
用javascript关闭本窗口技巧小结
2014/09/05 Javascript
AngularJS中如何使用$parse或$eval在运行时对Scope变量赋值
2016/01/25 Javascript
Angularjs CURD 详解及实例代码
2016/09/14 Javascript
JavaScript 闭包详细介绍
2016/09/28 Javascript
JavaScript实现类似拉勾网的鼠标移入移出效果
2016/10/27 Javascript
jQGrid动态填充select下拉框的选项值(动态填充)
2016/11/28 Javascript
vue.js开发环境安装教程
2017/03/17 Javascript
Angular2数据绑定详解
2017/04/18 Javascript
JS获取子、父、兄节点方法小结
2017/08/14 Javascript
AngularJS实现注册表单验证功能
2017/10/16 Javascript
Redux实现组合计数器的示例代码
2018/07/04 Javascript
详解滑动穿透(锁body)终极探索
2019/04/16 Javascript
js抽奖转盘实现方法分析
2020/05/16 Javascript
Python的subprocess模块总结
2014/11/07 Python
Java中重定向输出流实现用文件记录程序日志
2015/06/12 Python
Django中的CACHE_BACKEND参数和站点级Cache设置
2015/07/23 Python
python抓取网页内容并进行语音播报的方法
2018/12/24 Python
win10环境下配置vscode python开发环境的教程详解
2019/10/16 Python
Python中常用的高阶函数实例详解
2020/02/21 Python
护理自荐信
2013/10/22 职场文书
中医专业应届生求职信
2013/11/17 职场文书
自荐信如何制作?
2014/02/21 职场文书
就职演讲稿范文
2014/05/19 职场文书
大学生实习证明范文(5篇)
2014/09/18 职场文书
无犯罪记录证明
2014/09/19 职场文书
农民工工资支付承诺书
2015/05/04 职场文书
结婚喜宴迎宾词
2015/08/10 职场文书
2016新教师岗前培训心得体会
2016/01/08 职场文书
解决SpringCloud Feign传对象参数调用失败的问题
2021/06/23 Java/Android