在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实现协同过滤的教程
Apr 08 Python
python实现下载指定网址所有图片的方法
Aug 08 Python
详解python做UI界面的方法
Feb 27 Python
查看端口并杀进程python脚本代码
Dec 17 Python
解决tensorflow打印tensor有省略号的问题
Feb 04 Python
使用Python实现牛顿法求极值
Feb 10 Python
解决Django提交表单报错:CSRF token missing or incorrect的问题
Mar 13 Python
Python bisect模块原理及常见实例
Jun 17 Python
完美解决TensorFlow和Keras大数据量内存溢出的问题
Jul 03 Python
Django视图类型总结
Feb 17 Python
python操作xlsx格式文件并读取
Jun 02 Python
python随机打印成绩排名表
Jun 23 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
松下Panasonic RF-B65电路分析
2021/03/02 无线电
基于php socket(fsockopen)的应用实例分析
2013/06/02 PHP
Apache服务器下防止图片盗链的办法
2015/07/06 PHP
PHP实现递归无限级分类
2015/10/22 PHP
php 从一个数组中随机的取出若干个不同的数实例
2016/12/31 PHP
PHP生成(支持多模板)二维码海报代码
2018/04/30 PHP
浅谈PHP封装CURL
2019/03/06 PHP
JavaScript Object的extend是一个常用的功能
2009/12/02 Javascript
IE与firefox下Dhtml的一些区别小结
2009/12/02 Javascript
js showModalDialog参数的使用详解
2014/01/07 Javascript
JavaScript实现节点的删除与序号重建实例
2015/08/05 Javascript
JavaScript中日期的相关操作方法总结
2015/10/24 Javascript
解决JavaScript数字精度丢失问题的方法
2015/12/03 Javascript
Bootstrap实现提示框和弹出框效果
2017/01/11 Javascript
浅谈Webpack 持久化缓存实践
2018/03/22 Javascript
vuex进阶知识点巩固
2018/05/20 Javascript
node.js 模块和其下载资源的镜像设置的方法
2018/09/06 Javascript
vuejs+element UI点击编辑表格某一行时获取内容填入表单的示例
2018/10/31 Javascript
webpack4实现不同的导出类型
2019/04/09 Javascript
JavaScript键盘事件响应顺序详解
2019/09/30 Javascript
python实现TCP服务器端与客户端的方法详解
2015/04/30 Python
Python+PIL实现支付宝AR红包
2018/02/09 Python
Python3使用正则表达式爬取内涵段子示例
2018/04/22 Python
DataFrame:通过SparkSql将scala类转为DataFrame的方法
2019/01/29 Python
Python类的继承、多态及获取对象信息操作详解
2019/02/28 Python
TensorFlow基于MNIST数据集实现车牌识别(初步演示版)
2019/08/05 Python
python解析xml文件方式(解析、更新、写入)
2020/03/05 Python
CSS3模拟IOS滑动开关效果
2016/09/28 HTML / CSS
趣天网日本站:Qoo10 JP
2019/09/18 全球购物
解释DataSet(ds) 和 ds as DataSet 的含义
2014/07/27 面试题
劳资员岗位职责
2013/11/11 职场文书
班主任工作年限证明
2014/01/12 职场文书
领导党的群众路线教育实践活动个人对照检查材料
2014/09/23 职场文书
支行行长竞聘报告
2014/11/06 职场文书
投标承诺函范文
2015/01/21 职场文书
2016年员工政治思想表现评语
2015/12/02 职场文书