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发送邮件实例讲解(python发邮件附件可以使用email模块实现)
Dec 03 Python
python实现浪漫的烟花秀
Jan 30 Python
Python3.5模块的定义、导入、优化操作图文详解
Apr 27 Python
Python Matplotlib 基于networkx画关系网络图
Jul 10 Python
pytorch 实现模型不同层设置不同的学习率方式
Jan 06 Python
使用pytorch和torchtext进行文本分类的实例
Jan 08 Python
pytorch 模拟关系拟合——回归实例
Jan 14 Python
python实现秒杀商品的微信自动提醒功能(代码详解)
Apr 27 Python
python程序需要编译吗
Jun 19 Python
Python matplotlib图例放在外侧保存时显示不完整问题解决
Jul 28 Python
Django 用户认证Auth组件的使用
Nov 30 Python
Pytorch 实现变量类型转换
May 17 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&&mysql)六
2006/10/09 PHP
使用php将某个目录下面的所有文件罗列出来的方法详解
2013/06/21 PHP
php的declare控制符和ticks教程(附示例)
2014/03/21 PHP
如何在Web页面上直接打开、编辑、创建Office文档
2007/03/12 Javascript
jQuery 扩展对input的一些操作方法
2009/10/30 Javascript
jQuery get和post 方法传值注意事项
2009/11/03 Javascript
深入理解javascript学习笔记(一) 编写高质量代码
2012/08/09 Javascript
跟我学Nodejs(一)--- Node.js简介及安装开发环境
2014/05/20 NodeJs
Jquery中扩展方法extend使用技巧
2014/08/24 Javascript
javascript事件冒泡实例分析
2015/05/13 Javascript
基于JavaScript创建动态Dom
2015/12/08 Javascript
JavaScript Promise 用法
2016/06/14 Javascript
Angular+Bootstrap+Spring Boot实现分页功能实例代码
2017/07/21 Javascript
Redux 和 Mobx的选择问题:让你不再困惑!
2017/09/18 Javascript
jquery实现楼层滚动效果
2018/01/01 jQuery
vue中实现图片和文件上传的示例代码
2018/03/16 Javascript
JS简单判断是否在微信浏览器打开的方法示例
2019/01/08 Javascript
javascript面向对象三大特征之多态实例详解
2019/07/24 Javascript
Vue路由管理器Vue-router的使用方法详解
2020/02/05 Javascript
videocapture库制作python视频高速传输程序
2013/12/23 Python
朴素贝叶斯算法的python实现方法
2014/11/18 Python
介绍Python的Django框架中的静态资源管理器django-pipeline
2015/04/25 Python
python中metaclass原理与用法详解
2019/06/25 Python
Python urlencode和unquote函数使用实例解析
2020/03/31 Python
tensorflow转换ckpt为savermodel模型的实现
2020/05/25 Python
巴西在线鞋店:Shoestock
2017/10/28 全球购物
英国屋顶用品和材料超市:Roofing Supplies UK
2019/08/24 全球购物
商得四方公司面试题(gid+)
2014/04/30 面试题
行政经理的岗位职责
2013/11/23 职场文书
好家长事迹材料
2014/01/23 职场文书
汉语言文学毕业求职信
2014/07/17 职场文书
护士年终考核评语
2014/12/31 职场文书
亲属关系公证书样本
2015/01/23 职场文书
小班教师个人总结
2015/02/05 职场文书
因家庭原因离职的辞职信范文
2015/05/12 职场文书
教师节校长致辞
2015/07/31 职场文书