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编程中的字符串
Oct 14 Python
TensorFlow安装及jupyter notebook配置方法
Sep 08 Python
TensorFlow实现模型评估
Sep 07 Python
在Pycharm中对代码进行注释和缩进的方法详解
Jan 20 Python
图文详解python安装Scrapy框架步骤
May 20 Python
Django视图扩展类知识点详解
Oct 25 Python
Pytorch技巧:DataLoader的collate_fn参数使用详解
Jan 08 Python
python正则过滤字母、中文、数字及特殊字符方法详解
Feb 11 Python
Python单元测试及unittest框架用法实例解析
Jul 09 Python
Matlab中plot基本用法的具体使用
Jul 17 Python
Python实现数字的格式化输出
Aug 01 Python
详解python第三方库的安装、PyInstaller库、random库
Mar 03 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 中文处理函数集合
2008/08/27 PHP
PHP 作用域解析运算符(::)
2010/07/27 PHP
PHP生成各种常见验证码和Ajax验证过程
2016/01/10 PHP
Zend Framework框架Smarty扩展实现方法
2016/03/22 PHP
PHP 记录访客的浏览信息方法
2018/01/29 PHP
Yii框架日志操作图文与实例详解
2019/09/09 PHP
ThinkPHP类似AOP思想的参数验证的实现方法
2019/12/18 PHP
JavaScript小技巧 2.5 则
2010/09/12 Javascript
浏览器常用高宽的jquery插件
2011/02/24 Javascript
jQuery 插件开发指南
2014/11/14 Javascript
jQuery实现首页顶部可伸缩广告特效代码
2015/04/15 Javascript
JS+CSS实现的蓝色table选项卡效果
2015/10/08 Javascript
Bootstrap 最常用的JS插件系列总结(图片轮播、标签切换等)
2016/07/14 Javascript
ES6下子组件调用父组件的方法(推荐)
2018/02/23 Javascript
vue中实现图片和文件上传的示例代码
2018/03/16 Javascript
详解Node使用Puppeteer完成一次复杂的爬虫
2018/04/18 Javascript
react-intl实现React国际化多语言的方法
2020/09/27 Javascript
[02:39]DOTA2英雄基础教程 极限穿梭编织者
2013/12/05 DOTA
[05:28]刀塔密之一:团结则存
2014/07/03 DOTA
Python设计模式之抽象工厂模式
2016/08/25 Python
Python爬取京东的商品分类与链接
2016/08/26 Python
python判断字符串是否是json格式方法分享
2017/11/07 Python
用sqlalchemy构建Django连接池的实例
2019/08/29 Python
css3实现波纹特效、H5实现动态波浪效果
2018/01/31 HTML / CSS
路政管理专业推荐信
2013/11/11 职场文书
工商学院毕业生自荐信
2013/11/12 职场文书
微博营销计划书
2014/01/10 职场文书
2014端午节活动策划方案
2014/01/27 职场文书
路政管理毕业自荐书范文
2014/02/10 职场文书
颁奖典礼主持词
2014/03/25 职场文书
《黄山奇石》教学反思
2014/04/19 职场文书
职位说明书范文
2014/05/07 职场文书
四风查摆问题及整改措施
2014/10/10 职场文书
自主招生推荐信怎么写
2015/03/26 职场文书
CSS filter 有什么神奇用途
2021/05/25 HTML / CSS
Win11使用CAD卡顿或者致命错误怎么办?Win11无法正常使用CAD的解决方法
2022/07/23 数码科技