在Pytorch中计算自己模型的FLOPs方式
更新时间：2019年12月30日 09:53:17 作者：2Dposeliu
今天小编就为大家分享一篇在Pytorch中计算自己模型的FLOPs方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
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 代码速度更快,今天这篇文章将会解答大家心中的疑惑2023-09-09
这篇文章主要介绍了python Timer 类使用介绍的相关资料，帮助大家更好的理解和使用python iupdatable包，感兴趣的朋友可以了解下2020-12-12
这篇文章主要介绍了Python networkx中获取图的邻接矩阵方式，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2022-12-12
这篇文章主要介绍了Python测试模块doctest使用解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2019-08-08
介绍Python的Django框架中的静态资源管理器django-pipeline
这篇文章主要介绍了介绍Python的Django框架中的静态资源管理器django-pipeline,django-pipeline是一个开源项目,被用来处理css等静态文件,需要的朋友可以参考下2015-04-04
Pandas中的unique()和nunique()区别详解
Pandas中Series和DataFrame的两种数据类型中都有nunique()和unique()方法，本文详细的介绍了两者的区别，具有一定的参考价值，感兴趣的可以了解一下2022-08-08
散点图是指在回归分析中，数据点在直角坐标系平面上的分布图，散点图表示因变量随自变量而变化的大致趋势，据此可以选择合适的函数对数据点进行拟合。本文将利用Python Pyecharts实现散点图的绘制，需要的可以参考一下2022-06-06
这篇文章主要为大家分享了一个基于Python和Pygame实现的赛车小游戏，文中的示例代码讲解详细，对我们学习Python有一定帮助，需要的可以参考一下2022-04-04
这篇文章主要介绍了Pythonic版二分查找实现过程原理解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-08-08
这篇文章主要介绍了python用类实现文章敏感词的过滤方法示例，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2019-10-10
最新评论