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异常处理慎用“except:pass”建议
Apr 02 Python
使用PyV8在Python爬虫中执行js代码
Feb 16 Python
用pycharm开发django项目示例代码
Oct 24 Python
学习python的前途 python挣钱
Feb 27 Python
Python3 执行Linux Bash命令的方法
Jul 12 Python
python过滤中英文标点符号的实例代码
Jul 15 Python
python创建与遍历List二维列表的方法
Aug 16 Python
Python实现清理微信僵尸粉功能示例【基于itchat模块】
May 29 Python
详解pytorch中squeeze()和unsqueeze()函数介绍
Sep 03 Python
浅谈Selenium 控制浏览器的常用方法
Dec 04 Python
Python中Permission denied的解决方案
Apr 02 Python
Python机器学习之PCA降维算法详解
May 19 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使用mb_check_encoding检查字符串在指定的编码里是否有效
2013/11/07 PHP
PHP迭代器实现斐波纳契数列的函数
2013/11/12 PHP
Linux中为php配置伪静态
2014/12/17 PHP
php curl上传、下载、https登陆实现代码
2017/07/23 PHP
php表单处理操作
2017/11/16 PHP
比较简单的异步加载JS文件的代码
2009/07/18 Javascript
jQuery插件 tabBox实现代码
2010/02/09 Javascript
Javascript 设计模式(二) 闭包
2010/05/26 Javascript
JQuery入门——事件切换之toggle()方法应用介绍
2013/02/05 Javascript
js动态创建及移除div的方法
2015/06/03 Javascript
微信小程序 传值取值的几种方法总结
2017/01/16 Javascript
jQuery实现的页面详情展开收起功能示例
2018/06/11 jQuery
三分钟学会用ES7中的Async/Await进行异步编程
2018/06/14 Javascript
一步步教你利用Docker设置Node.js
2018/11/20 Javascript
js实现简单的点名器随机色实例代码
2020/09/20 Javascript
python list 合并连接字符串的方法
2013/03/09 Python
Python实现的一个找零钱的小程序代码分享
2014/08/25 Python
使用Python在Windows下获取USB PID&VID的方法
2019/07/02 Python
mac使用python识别图形验证码功能
2020/01/10 Python
Python print不能立即打印的解决方式
2020/02/19 Python
利用django model save方法对未更改的字段依然进行了保存
2020/03/28 Python
Html5让容器充满屏幕高度或自适应剩余高度的布局实现
2020/05/14 HTML / CSS
科茨沃尔德家居商店:Scotts of Stow
2018/06/29 全球购物
函数只定义了一次, 调用了一次, 但编译器提示非法重定义了-什么问题?
2014/10/03 面试题
建筑实习自我鉴定
2013/10/18 职场文书
公务员年总结的自我评价
2013/10/25 职场文书
法学毕业生自我鉴定
2013/11/08 职场文书
大四学年自我鉴定
2013/11/13 职场文书
土木工程专业推荐信
2014/02/19 职场文书
2014年党员自我评价材料
2014/09/22 职场文书
2014年个人售房协议书
2014/10/30 职场文书
2014小学语文教师个人工作总结
2014/12/03 职场文书
通知函格式范文
2015/04/27 职场文书
农民工工资保障承诺书
2015/05/04 职场文书
详解如何在Canvas中添加事件的方法
2021/04/17 Javascript
Python 的 sum() Pythonic 的求和方法详细
2021/10/16 Python