在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根据经纬度计算距离示例
Feb 16 Python
Python线程中对join方法的运用的教程
Apr 09 Python
Pycharm学习教程(5) Python快捷键相关设置
May 03 Python
Python数据结构与算法之二叉树结构定义与遍历方法详解
Dec 12 Python
利用arcgis的python读取要素的X,Y方法
Dec 22 Python
python3实现指定目录下文件sha256及文件大小统计
Feb 25 Python
Python的高阶函数用法实例分析
Apr 11 Python
python自动化实现登录获取图片验证码功能
Nov 20 Python
python:目标检测模型预测准确度计算方式(基于IoU)
Jan 18 Python
在Tensorflow中查看权重的实现
Jan 24 Python
python_mask_array的用法
Feb 18 Python
没编程基础可以学python吗
Jun 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
smarty的保留变量问题
2008/10/23 PHP
CodeIgniter启用缓存和清除缓存的方法
2014/06/12 PHP
PHP获取用户客户端真实IP的解决方案
2016/10/10 PHP
Iframe 自适应高度并实时监控高度变化的js代码
2009/10/30 Javascript
jQuery 开天辟地入门篇一
2009/12/09 Javascript
基于jquery的9行js轻松实现tab控件示例
2013/10/12 Javascript
nodejs事件的监听与触发的理解分析
2015/02/12 NodeJs
实例讲解js验证表单项是否为空的方法
2016/01/09 Javascript
js制作网站首页图片轮播特效代码
2016/08/30 Javascript
JS实现可编辑的后台管理菜单功能【附demo源码下载】
2016/09/13 Javascript
easyui messager alert 三秒后自动关闭提示的实例
2016/11/07 Javascript
详解用原生JavaScript实现jQuery的某些简单功能
2016/12/19 Javascript
HTML5 js实现拖拉上传文件功能
2020/11/20 Javascript
使用Node.js搭建静态资源服务详细教程
2017/08/02 Javascript
react.js使用webpack搭配环境的入门教程
2017/08/14 Javascript
解决VUE框架 导致绑定事件的阻止冒泡失效问题
2018/02/24 Javascript
webpack配置打包后图片路径出错的解决
2018/04/26 Javascript
vue.js实现三级菜单效果
2019/10/19 Javascript
JS前端基于canvas给图片添加水印
2020/11/11 Javascript
python实现dict版图遍历示例
2014/02/19 Python
Python ORM框架SQLAlchemy学习笔记之数据添加和事务回滚介绍
2014/06/10 Python
Python实现堆排序的方法详解
2016/05/03 Python
200行自定义python异步非阻塞Web框架
2017/03/15 Python
python定向爬取淘宝商品价格
2018/02/27 Python
python利用小波分析进行特征提取的实例
2019/01/09 Python
django中账号密码验证登陆功能的实现方法
2019/07/15 Python
python如何将多个PDF进行合并
2019/08/13 Python
python中plt.imshow与cv2.imshow显示颜色问题
2020/07/16 Python
详解HTML5中ol标签的用法
2015/09/08 HTML / CSS
医学检验专业个人求职信范文
2013/12/04 职场文书
优秀中学生事迹材料
2014/01/31 职场文书
班主任工作实习计划
2015/01/16 职场文书
开学典礼校长致辞
2015/07/29 职场文书
基于python制作简易版学生信息管理系统
2021/04/20 Python
Python爬取英雄联盟MSI直播间弹幕并生成词云图
2021/06/01 Python
Go 语言下基于Redis分布式锁的实现方式
2021/06/28 Golang