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中的XML库4Suite Server的介绍
Apr 14 Python
python+selenium实现京东自动登录及秒杀功能
Nov 18 Python
使用Python进行QQ批量登录的实例代码
Jun 11 Python
python 从csv读数据到mysql的实例
Jun 21 Python
简单谈谈python基本数据类型
Sep 26 Python
Python字符串的常见操作实例小结
Apr 08 Python
python实现抠图给证件照换背景源码
Aug 20 Python
使用OpenCV实现仿射变换—平移功能
Aug 29 Python
查看jupyter notebook每个单元格运行时间实例
Apr 22 Python
解决Keras中CNN输入维度报错问题
Jun 29 Python
Django ModelForm组件原理及用法详解
Oct 12 Python
Python os和os.path模块详情
Apr 02 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
php+mysql写的简单留言本实例代码
2008/07/25 PHP
PHP自带方法验证邮箱、URL、IP是否合法的函数
2016/12/08 PHP
CentOS7系统搭建LAMP及更新PHP版本操作详解
2020/03/26 PHP
JS禁用浏览器退格键实现思路及代码
2013/10/29 Javascript
JS完整获取IE浏览器信息包括类型、版本、语言等等
2014/05/22 Javascript
jQuery中:nth-child选择器用法实例
2014/12/31 Javascript
javascript跨域方法、原理以及出现问题解决方法(详解)
2015/08/06 Javascript
AngularJS实现Model缓存的方式
2016/02/03 Javascript
详解XMLHttpRequest(二)响应属性、二进制数据、监测上传下载进度
2016/09/14 Javascript
JS实现的几个常用算法
2016/11/12 Javascript
JS生成和下载二维码的代码
2016/12/07 Javascript
AngularJS通过ng-Img-Crop实现头像截取的示例
2017/08/17 Javascript
详解RequireJs官方使用教程
2017/10/31 Javascript
webpack打包js的方法
2018/03/12 Javascript
vue如何将v-for中的表格导出来
2018/05/07 Javascript
Bootstrap开发中Tab标签页切换图表显示问题的解决方法
2018/07/13 Javascript
详解ES6 Promise的生命周期和创建
2019/08/18 Javascript
element-ui中按需引入的实现
2019/12/25 Javascript
使用python BeautifulSoup库抓取58手机维修信息
2013/11/21 Python
在Python web中实现验证码图片代码分享
2017/11/09 Python
Python实现的KMeans聚类算法实例分析
2018/12/29 Python
python pyinstaller 加载ui路径方法
2019/06/10 Python
python命令行参数用法实例分析
2019/06/25 Python
python中delattr删除对象方法的代码分析
2020/12/15 Python
pycharm配置QtDesigner的超详细方法
2021/01/25 Python
HTML5拖拽的简单实例
2016/05/30 HTML / CSS
空指针到底是什么
2012/08/07 面试题
当当网软件测试笔试题
2015/11/24 面试题
如何开发安全的AJAX应用
2014/03/26 面试题
市场推广策划方案
2014/06/02 职场文书
局领导领导班子四风对照检查材料
2014/09/27 职场文书
党的群众路线教育实践活动教师自我剖析材料
2014/10/09 职场文书
语文复习计划
2015/01/19 职场文书
党性修养心得体会2016
2016/01/21 职场文书
python 三边测量定位的实现代码
2021/04/22 Python
java设计模式--七大原则详解
2021/07/21 Java/Android