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 28 Python
Windows下Python使用Pandas模块操作Excel文件的教程
May 31 Python
python逐行读写txt文件的实例讲解
Apr 03 Python
Python基于多线程操作数据库相关问题分析
Jul 11 Python
使用Python和Scribus创建一个RGB立方体的方法
Jul 17 Python
python爬取盘搜的有效链接实现代码
Jul 20 Python
Python3 Tkinkter + SQLite实现登录和注册界面
Nov 19 Python
win10下python2和python3共存问题解决方法
Dec 23 Python
python matplotlib画盒图、子图解决坐标轴标签重叠的问题
Jan 19 Python
Python开发之身份证验证库id_validator验证身份证号合法性及根据身份证号返回住址年龄等信息
Mar 20 Python
Python二元算术运算常用方法解析
Sep 15 Python
Python爬取用户观影数据并分析用户与电影之间的隐藏信息!
Jun 29 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面试题附答案
2009/01/07 PHP
php中经典方法实现判断多维数组是否为空
2011/10/23 PHP
掌握PHP垃圾回收机制详解
2019/03/13 PHP
Laravel基础_关于view共享数据的示例讲解
2019/10/14 PHP
Laravel 实现在Blade模版中使用全局变量代替路径的例子
2019/10/22 PHP
JavaScript入门教程(9) Document文档对象
2009/01/31 Javascript
jQuery-Easyui 1.2 实现多层菜单效果的代码
2012/01/13 Javascript
在JS数组特定索引处指定位置插入元素的技巧
2014/08/24 Javascript
jquery动态改变div宽度和高度
2015/02/09 Javascript
JavaScript+CSS实现仿Mootools竖排弹性动画菜单效果
2015/10/14 Javascript
深入分析Javascript事件代理
2016/01/30 Javascript
angularJS Provider、factory、service详解及实例代码
2016/09/21 Javascript
类似于QQ的右滑删除效果的实现方法
2016/10/16 Javascript
AngularJS中指令的四种基本形式实例分析
2016/11/22 Javascript
bootstrap jquery dataTable 异步ajax刷新表格数据的实现方法
2017/02/10 Javascript
浅谈jQuery中的$.extend方法来扩展JSON对象
2017/02/12 Javascript
浅谈Vue2.0中v-for迭代语法的变化(key、index)
2018/03/06 Javascript
javaScript实现鼠标在文字上悬浮时弹出悬浮层效果
2020/04/12 Javascript
js的对象与函数详解
2019/01/21 Javascript
微信小程序MUI导航栏透明渐变功能示例(通过改变rgba的a值实现)
2019/01/24 Javascript
vue 引用自定义ttf、otf、在线字体的方法
2019/05/09 Javascript
JavaScript正则表达式验证登录实例
2020/03/18 Javascript
Python写的Tkinter程序屏幕居中方法
2015/03/10 Python
python读写ini配置文件方法实例分析
2015/06/30 Python
基于python的七种经典排序算法(推荐)
2016/12/08 Python
Python实现将照片变成卡通图片的方法【基于opencv】
2018/01/17 Python
Windows下python3安装tkinter的问题及解决方法
2020/01/06 Python
win10安装tesserocr配置 Python使用tesserocr识别字母数字验证码
2020/01/16 Python
keras-siamese用自己的数据集实现详解
2020/06/10 Python
东南亚旅游平台:The Trip Guru
2018/01/01 全球购物
类如何去实现接口
2013/12/19 面试题
实现strstr功能,即在父串中寻找子串首次出现的位置
2016/08/05 面试题
将一个文本文件的内容按倒序打印出来
2015/01/05 面试题
经管应届生求职信范文
2014/05/18 职场文书
毕业实习指导教师评语
2014/12/31 职场文书
漫改真人电影「萌系男友是燃燃的橘色」公开先导视觉图
2022/03/21 日漫