在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获取网页状态码示例
Mar 30 Python
Python元组操作实例分析【创建、赋值、更新、删除等】
Jul 24 Python
python数字图像处理之高级形态学处理
Apr 27 Python
Python随机生成身份证号码及校验功能
Dec 04 Python
python实现视频分帧效果
May 31 Python
Python 实现Numpy中找出array中最大值所对应的行和列
Nov 26 Python
Python实现AI换脸功能
Apr 10 Python
Python实现EM算法实例代码
Oct 04 Python
python time()的实例用法
Nov 03 Python
python 逐步回归算法
Apr 06 Python
Keras在mnist上的CNN实践,并且自定义loss函数曲线图操作
May 25 Python
Pygame如何使用精灵和碰撞检测
Nov 17 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
ThinkPHP模板判断输出Present标签用法详解
2014/06/30 PHP
php实现图片文件与下载文件防盗链的方法
2014/11/03 PHP
php实现在服务器上创建目录的方法
2015/03/16 PHP
Bootstrap+PHP实现多图上传功能实例详解
2018/04/08 PHP
thinkphp5 框架结合plupload实现图片批量上传功能示例
2020/04/04 PHP
ExtJS 2.2.1的grid控件在ie6中的显示问题
2009/05/04 Javascript
JQuery 选择和过滤方法代码总结
2010/11/19 Javascript
JS target与currentTarget区别说明
2011/08/28 Javascript
js写的评论分页(还不错)
2013/12/23 Javascript
javascript实现锁定网页、密码解锁效果(类似系统屏幕保护效果)
2014/08/15 Javascript
jquery使用each方法遍历json格式数据实例
2015/05/18 Javascript
jQuery Html控件基本操作(日常收集整理)
2016/03/11 Javascript
微信小程序通过api接口将json数据展现到小程序示例
2017/01/20 Javascript
微信小程序实现图片自适应(支持多图)
2017/01/25 Javascript
JS中去掉array中重复元素的方法
2017/05/26 Javascript
继承行为在 ES5 与 ES6 中的区别详解
2019/12/24 Javascript
简单了解vue 插值表达式Mustache
2020/07/22 Javascript
Python中if __name__ == '__main__'作用解析
2015/06/29 Python
python pands实现execl转csv 并修改csv指定列的方法
2018/12/12 Python
Pycharm安装并配置jupyter notebook的实现
2020/05/18 Python
Django日志及中间件模块应用案例
2020/09/10 Python
python实现MySQL指定表增量同步数据到clickhouse的脚本
2021/02/26 Python
美国标志性加大尺码时装品牌:Ashley Stewart
2016/12/15 全球购物
下面这个程序执行后会有什么错误或者效果
2014/11/03 面试题
大专生工程监理求职信
2013/10/04 职场文书
求职简历自我评价范例
2014/03/12 职场文书
《陈毅探母》教学反思
2014/05/01 职场文书
热爱祖国演讲稿
2014/05/04 职场文书
群众路线教育实践活动方案
2014/10/31 职场文书
二手车交易协议书标准版
2014/11/16 职场文书
幼儿园校车安全责任书
2015/05/08 职场文书
民事起诉状范文
2015/05/19 职场文书
2016同学毕业寄语大全
2015/12/04 职场文书
廉洁自律心得体会2016
2016/01/13 职场文书
看完这篇文章获得一些java if优化技巧
2021/07/15 Java/Android
Nginx 反向代理解决跨域问题多种情况分析
2022/01/18 Servers