在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中__call__内置函数用法实例
Jun 04 Python
Bottle框架中的装饰器类和描述符应用详解
Oct 28 Python
python实现超市扫码仪计费
May 30 Python
python实现requests发送/上传多个文件的示例
Jun 04 Python
Python中的random.uniform()函数教程与实例解析
Mar 02 Python
快速排序的四种python实现(推荐)
Apr 03 Python
Python Flask框架模板操作实例分析
May 03 Python
Python三元运算与lambda表达式实例解析
Nov 30 Python
python字符串替换re.sub()实例解析
Feb 09 Python
Anconda环境下Vscode安装Python的方法详解
Mar 29 Python
python各种excel写入方式的速度对比
Nov 10 Python
python使用scapy模块实现ARP扫描的过程
Jan 21 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
深入探讨PHP中的内存管理问题
2011/08/31 PHP
php中的strpos使用示例
2014/02/27 PHP
php实现的日历程序
2015/06/18 PHP
php中yar框架实例用法讲解
2020/12/27 PHP
input的focus方法使用
2010/03/13 Javascript
jquery下将选择的checkbox的id组成字符串的方法
2010/11/28 Javascript
基于JavaScript实现继承机制之原型链(prototype chaining)的详解
2013/05/07 Javascript
只需一行代码,轻松实现一个在线编辑器
2013/11/12 Javascript
js获取当月最后一天实例代码
2013/11/19 Javascript
angularjs实现与服务器交互分享
2014/06/24 Javascript
JS仿Windows开机启动Loading进度条的方法
2015/02/26 Javascript
iframe里使用JavaScript控制主页转向的方法
2015/04/03 Javascript
实例讲解javascript注册事件处理函数
2016/01/09 Javascript
js类式继承与原型式继承详解
2016/04/07 Javascript
微信小程序 template模板详解及实例
2017/02/21 Javascript
JS实现的tab切换选项卡效果示例
2017/02/28 Javascript
vue2.0在table中实现全选和反选的示例代码
2017/11/04 Javascript
js传递数组参数到后台controller的方法
2018/03/29 Javascript
解决vue-cli项目打包出现空白页和路径错误的问题
2018/09/04 Javascript
javascript实现小型区块链功能
2019/04/03 Javascript
vue + any-touch实现一个iscroll 实现拖拽和滑动动画效果
2019/04/08 Javascript
Element Notification通知的实现示例
2020/07/27 Javascript
[03:08]迎霜节狂欢!2018年迎霜节珍藏Ⅰ一览
2018/12/25 DOTA
python @property的用法及含义全面解析
2018/02/01 Python
pyqt5 键盘监听按下enter 就登陆的实例
2019/06/25 Python
django多个APP的urls设置方法(views重复问题解决)
2019/07/19 Python
parser.add_argument中的action使用
2020/04/20 Python
澳洲国民品牌乡村路折扣店:Country Road & Trenery Outlet
2018/04/19 全球购物
eBay奥地利站:eBay.at
2019/07/24 全球购物
异步传递消息系统的作用
2016/05/01 面试题
写自荐信有哪些不宜?
2013/10/17 职场文书
高中生期末评语
2014/01/28 职场文书
《诺贝尔》教学反思
2014/02/17 职场文书
生产厂长岗位职责
2014/02/21 职场文书
教师党的群众路线教育实践活动剖析材料
2014/10/09 职场文书
公安民警正风肃纪剖析材料
2014/10/10 职场文书