PyTorch和Keras计算模型参数的例子


Posted in Python onJanuary 02, 2020

Pytorch中,变量参数,用numel得到参数数目,累加

def get_parameter_number(net):
  total_num = sum(p.numel() for p in net.parameters())
  trainable_num = sum(p.numel() for p in net.parameters() if p.requires_grad)
  return {'Total': total_num, 'Trainable': trainable_num}

Keras中,直接使用model的summary函数

model = k_model()
model.summary()

以上这篇PyTorch和Keras计算模型参数的例子就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python中的下划线详解
Jun 24 Python
在Python的Django框架中创建语言文件
Jul 27 Python
Python中规范定义命名空间的一些建议
Jun 04 Python
matplotlib简介,安装和简单实例代码
Dec 26 Python
Django REST framework视图的用法
Jan 16 Python
python图像处理入门(一)
Apr 04 Python
pycharm创建一个python包方法图解
Apr 10 Python
python3.6使用tkinter实现弹跳小球游戏
May 09 Python
机器学习实战之knn算法pandas
Jun 22 Python
Python是什么 Python的用处
May 26 Python
python中如何打包用户自定义模块
Sep 23 Python
pytorch 梯度NAN异常值的解决方案
Jun 05 Python
Pytorch中实现只导入部分模型参数的方式
Jan 02 #Python
PyTorch中topk函数的用法详解
Jan 02 #Python
Pytorch训练过程出现nan的解决方式
Jan 02 #Python
pytorch绘制并显示loss曲线和acc曲线,LeNet5识别图像准确率
Jan 02 #Python
基于MSELoss()与CrossEntropyLoss()的区别详解
Jan 02 #Python
python使用SQLAlchemy操作MySQL
Jan 02 #Python
pytorch 实现cross entropy损失函数计算方式
Jan 02 #Python
You might like
解析crontab php自动运行的方法
2013/06/24 PHP
php堆排序实现原理与应用方法
2015/01/03 PHP
ThinkPHP实现登录退出功能
2017/06/29 PHP
PHP封装请求类实例分析【基于Yii框架】
2019/10/17 PHP
基于jQuery实现点击同时更改两个iframe的网址
2010/07/01 Javascript
javascript中callee与caller的用法和应用场景
2010/12/08 Javascript
Struts2的s:radio标签使用及用jquery添加change事件
2013/04/08 Javascript
教你在heroku云平台上部署Node.js应用
2014/07/30 Javascript
2014 年最热门的21款JavaScript框架推荐
2014/12/25 Javascript
javascript面向对象程序设计(一)
2015/01/29 Javascript
javascript实现自动填写表单实例简析
2015/12/02 Javascript
XML、HTML、CSS与JS的区别整理
2016/02/18 Javascript
bootstrap精简教程_动力节点Java学院整理
2017/07/14 Javascript
JS实现简易换图时钟功能分析
2018/01/04 Javascript
vue-cli常用设置总结
2018/02/24 Javascript
解决iView中时间控件选择的时间总是少一天的问题
2018/03/15 Javascript
vue-infinite-loading2.0 中文文档详解
2018/04/08 Javascript
egg.js的基本使用和调用数据库的方法示例
2019/05/18 Javascript
javascript实现简易计算器功能
2020/09/23 Javascript
[02:37]2018DOTA2亚洲邀请赛赛前采访 VP.no[o]ne心中最强SOLO是谁
2018/04/04 DOTA
[32:47]完美世界DOTA2联赛 GXR vs IO 第二场 11.07
2020/11/09 DOTA
influx+grafana自定义python采集数据和一些坑的总结
2018/09/17 Python
python实现反转部分单向链表
2018/09/27 Python
Pandas DataFrame 取一行数据会得到Series的方法
2018/11/10 Python
Pytorch使用MNIST数据集实现CGAN和生成指定的数字方式
2020/01/10 Python
tensorflow模型保存、加载之变量重命名实例
2020/01/21 Python
python爬取王者荣耀全皮肤的简单实现代码
2020/01/31 Python
python GUI框架pyqt5 对图片进行流式布局的方法(瀑布流flowlayout)
2020/03/12 Python
python能在浏览器能运行吗
2020/06/17 Python
Python RabbitMQ实现简单的进程间通信示例
2020/07/02 Python
西班牙多品牌鞋店连锁店:Krack
2018/11/30 全球购物
人力管理专业毕业生求职信
2014/02/27 职场文书
高中教师考核方案
2014/05/18 职场文书
学雷锋活动简报
2015/07/20 职场文书
小学六一儿童节活动开幕词
2016/03/04 职场文书
Python OpenCV 彩色与灰度图像的转换实现
2021/06/05 Python