在Pytorch中计算自己模型的FLOPs方式


Posted in Python onDecember 30, 2019

https://github.com/Lyken17/pytorch-OpCounter

安装方法很简单:

pip install thop

基本用法:

from torchvision.models import resnet50from thop import profile
model = resnet50()
flops, params = profile(model, input_size=(1, 3, 224,224))

对自己的module进行特别的计算:

class YourModule(nn.Module):
# your definition
def count_your_model(model, x, y):
# your rule
hereflops, params = profile(model, input_size=(1, 3, 224,224),
custom_ops={YourModule: count_your_model})

以上这篇在Pytorch中计算自己模型的FLOPs方式就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
跟老齐学Python之让人欢喜让人忧的迭代
Oct 02 Python
Python中用于计算对数的log()方法
May 15 Python
深入分析python数据挖掘 Json结构分析
Apr 21 Python
Python匿名函数及应用示例
Apr 09 Python
Python利用pandas处理Excel数据的应用详解
Jun 18 Python
windows上安装python3教程以及环境变量配置详解
Jul 18 Python
通过python扫描二维码/条形码并打印数据
Nov 14 Python
python进程的状态、创建及使用方法详解
Dec 06 Python
如何基于python把文字图片写入word文档
Jul 31 Python
Python中seaborn库之countplot的数据可视化使用
Jun 11 Python
Django模型层实现多表关系创建和多表操作
Jul 21 Python
Python基本知识点总结
Apr 07 Python
Pytorch之保存读取模型实例
Dec 30 #Python
Python爬虫解析网页的4种方式实例及原理解析
Dec 30 #Python
Python中如何将一个类方法变为多个方法
Dec 30 #Python
pytorch 实现打印模型的参数值
Dec 30 #Python
Python如何基于smtplib发不同格式的邮件
Dec 30 #Python
pytorch获取模型某一层参数名及参数值方式
Dec 30 #Python
Python类反射机制使用实例解析
Dec 30 #Python
You might like
通过ODBC连接的SQL SERVER实例
2006/10/09 PHP
windows下PHP APACHE MYSQ完整配置
2007/01/02 PHP
追求程序速度,而不是编程的速度
2008/04/23 PHP
php 高效率写法 推荐
2010/02/21 PHP
php抓取页面与代码解析 推荐
2010/07/23 PHP
比较discuz和ecshop的截取字符串函数php版
2012/09/03 PHP
PHP callback函数使用方法和注意事项
2015/01/23 PHP
使用PHP和JavaScript判断请求是否来自微信内浏览器
2015/08/18 PHP
php通过会话控制实现身份验证实例
2016/10/18 PHP
Dom与浏览器兼容性说明
2010/10/25 Javascript
JavaScript EasyPager 分页函数
2011/05/25 Javascript
基于dom编程中 动态创建与删除元素的使用
2013/04/17 Javascript
Jquery同辈元素选中/未选中效果的实例代码
2013/08/01 Javascript
IE、FF、Chrome浏览器中的JS差异介绍
2013/08/13 Javascript
IE下Ajax缓存问题的快速解决方法(get方式)
2014/01/09 Javascript
Javascript实现快速排序(Quicksort)的算法详解
2015/09/06 Javascript
JQuery实现简单的图片滑动切换特效
2015/11/22 Javascript
浅谈js中startsWith 函数不能在任何浏览器兼容的问题
2017/03/01 Javascript
如何写好你的JavaScript【推荐】
2017/03/02 Javascript
JS验证全角与半角及相互转化的介绍
2017/05/18 Javascript
[js高手之路]HTML标签解释成DOM节点的实现方法
2017/08/31 Javascript
JS从非数组对象转数组的方法小结
2018/03/26 Javascript
用Cordova打包Vue项目的方法步骤
2019/02/02 Javascript
JavaScript封闭函数及常用内置对象示例
2019/05/13 Javascript
JS监听组合按键思路及实现过程
2020/04/17 Javascript
py2exe 编译ico图标的代码
2013/03/08 Python
浅谈python脚本设置运行参数的方法
2018/12/03 Python
Python3+OpenCV2实现图像的几何变换(平移、镜像、缩放、旋转、仿射)
2019/05/13 Python
python模拟斗地主发牌
2020/04/22 Python
python golang中grpc 使用示例代码详解
2020/06/03 Python
CSS3 重置iphone浏览器按钮input,select等表单元素的默认样式
2014/10/11 HTML / CSS
企业承诺书怎么写
2014/05/24 职场文书
团队拓展活动总结
2014/08/27 职场文书
放飞理想演讲稿
2014/09/09 职场文书
作风大整顿心得体会
2014/09/10 职场文书
领导班子“四风问题”“整改方案
2014/10/02 职场文书