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使用urllib模块开发的多线程豆瓣小站mp3下载器
Jan 16 Python
从零学python系列之数据处理编程实例(二)
May 22 Python
python中的hashlib和base64加密模块使用实例
Sep 02 Python
Python中类型关系和继承关系实例详解
May 25 Python
python if not in 多条件判断代码
Sep 21 Python
浅谈flask截获所有访问及before/after_request修饰器
Jan 18 Python
Python Xml文件添加字节属性的方法
Mar 31 Python
Python 函数用法简单示例【定义、参数、返回值、函数嵌套】
Sep 20 Python
pytorch 实现将自己的图片数据处理成可以训练的图片类型
Jan 08 Python
详解用python -m http.server搭一个简易的本地局域网
Sep 24 Python
python geopandas读取、创建shapefile文件的方法
Jun 29 Python
Python 数据可视化之Matplotlib详解
Nov 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数组 为文章加关键字连接 文章内容自动加链接
2011/12/29 PHP
php验证手机号码
2015/11/11 PHP
JS获取页面窗口大小的代码解读
2011/12/01 Javascript
js函数名与form表单元素同名冲突的问题
2014/03/07 Javascript
跟我学Nodejs(三)--- Node.js模块
2014/05/25 NodeJs
JavaScript中数据结构与算法(三):链表
2015/06/19 Javascript
jquery实现的3D旋转木马特效代码分享
2015/08/25 Javascript
JavaScript 轮播图和自定义滚动条配合鼠标滚轮分享代码贴
2016/10/28 Javascript
JavaScript模板引擎Template.js使用详解
2016/12/15 Javascript
footer定位页面底部(代码分享)
2017/03/07 Javascript
vue中将网页打印成pdf实例代码
2017/06/15 Javascript
分享JS表单验证源码(带错误提示及密码等级)
2020/01/05 Javascript
JS实现iframe中子父页面跨域通讯的方法分析
2020/03/10 Javascript
vue和小程序项目中使用iconfont的方法
2020/05/19 Javascript
浅谈Vue static 静态资源路径 和 style问题
2020/11/07 Javascript
Python运行报错UnicodeDecodeError的解决方法
2016/06/07 Python
浅谈Python 对象内存占用
2016/07/15 Python
Python实现MySQL操作的方法小结【安装,连接,增删改查等】
2017/07/12 Python
Python 实现引用其他.py文件中的类和类的方法
2018/04/29 Python
Python selenium抓取微博内容的示例代码
2018/05/17 Python
使用Python对微信好友进行数据分析
2018/06/27 Python
python 输出所有大小写字母的方法
2019/01/02 Python
FFT快速傅里叶变换的python实现过程解析
2019/10/21 Python
简单html5代码获取地理位置
2014/03/31 HTML / CSS
HTML5 表单验证失败的提示语问题
2017/07/13 HTML / CSS
高级工程师岗位职责
2013/12/15 职场文书
单位消防安全制度
2014/01/12 职场文书
作弊检讨书1000字
2014/02/01 职场文书
环卫工人先进事迹材料
2014/06/02 职场文书
项目合作意向书模板
2014/07/29 职场文书
2014年国庆节庆祝建国65周年比赛演讲稿
2014/09/21 职场文书
大四学生个人总结
2015/02/15 职场文书
2015年后勤工作总结范文
2015/04/08 职场文书
培训后的感想
2015/08/07 职场文书
SQLServer2019 数据库环境搭建与使用的实现
2021/04/08 SQL Server
Python基础之hashlib模块详解
2021/05/06 Python