在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聊天程序实例代码分享
Nov 18 Python
python实现删除文件与目录的方法
Nov 10 Python
Django中利用filter与simple_tag为前端自定义函数的实现方法
Jun 15 Python
python3操作mysql数据库的方法
Jun 23 Python
浅谈Python黑帽子取代netcat
Feb 10 Python
Python文本处理之按行处理大文件的方法
Apr 09 Python
python3爬虫之设计签名小程序
Jun 19 Python
django利用request id便于定位及给日志加上request_id
Aug 26 Python
Python一个简单的通信程序(客户端 服务器)
Mar 06 Python
python使用requests模块实现爬取电影天堂最新电影信息
Apr 03 Python
python urllib爬虫模块使用解析
Sep 05 Python
详解Django配置JWT认证方式
May 09 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 删除cookie和浏览器重定向
2009/03/16 PHP
解析PHP留言本模块主要功能的函数说明(代码可实现)
2013/06/25 PHP
ThinkPHP中__initialize()和类的构造函数__construct()用法分析
2014/11/29 PHP
php 函数使用可变数量的参数方法
2017/05/02 PHP
PHP结合Ffmpeg快速搭建流媒体服务的实践记录
2018/10/31 PHP
laravel 数据迁移与 Eloquent ORM的实现方法
2019/04/12 PHP
jQuery学习4 浏览器的事件模型
2010/02/07 Javascript
一个关于jqGrid使用的小例子(行按钮)
2011/11/04 Javascript
查看图片(前进后退)功能实现js代码
2013/04/24 Javascript
jQuery中first()方法用法实例
2015/01/06 Javascript
jQuery控制网页打印指定区域的方法
2015/04/07 Javascript
AngularJS入门教程之AngularJS表达式
2016/04/18 Javascript
js验证真实姓名与身份证号,手机号的简单实例
2016/07/18 Javascript
本地Bootstrap文件字体图标引入却无法显示问题的解决方法
2020/04/18 Javascript
JavaScript对象封装的简单实现方法(3种方法)
2017/01/03 Javascript
js模态对话框使用方法详解
2017/02/16 Javascript
self.attachevent is not a function的解决方法
2017/04/04 Javascript
weui框架实现上传、预览和删除图片功能代码
2017/08/24 Javascript
js求数组中全部数字可拼接出的最大整数示例代码
2017/08/25 Javascript
微信小程序自动客服功能
2017/11/02 Javascript
记一次Vue.js混入mixin的使用(分权限管理页面)
2019/04/17 Javascript
如何实现双向绑定mvvm的原理实现
2019/05/28 Javascript
layer 刷新某个页面的实现方法
2019/09/05 Javascript
vue中defineProperty和Proxy的区别详解
2020/11/30 Vue.js
[42:34]VP vs VG 2018国际邀请赛小组赛BO2 第一场 8.19
2018/08/21 DOTA
跟老齐学Python之复习if语句
2014/10/02 Python
python端口扫描系统实现方法
2014/11/19 Python
基于Keras的格式化输出Loss实现方式
2020/06/17 Python
python math模块的基本使用教程
2021/01/16 Python
大学本科毕业生的自我鉴定范文
2013/11/19 职场文书
社区护士演讲稿
2014/08/27 职场文书
2015社区六五普法工作总结
2015/04/21 职场文书
学校教师师德师风承诺书
2015/04/28 职场文书
2015大学生暑假调查报告
2015/07/13 职场文书
2016年清明节红领巾广播稿
2015/12/17 职场文书
从零开始在Centos7上部署SpringBoot项目
2022/04/07 Servers