使用thop库计算flops和params

最新推荐文章于 2024-07-03 11:12:09 发布
最新推荐文章于 2024-07-03 11:12:09 发布
阅读量219 收藏
点赞数
分类专栏： 深度学习 文章标签： 深度学习 计算机视觉 人工智能
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/qq_52396030/article/details/133017927
版权

from thop import profile
from thop import clever_format

input_data = torch.randn(1, 3, 256, 256).cuda()

flops, params = profile(model, inputs=(input_data,))
flops, params = clever_format([flops, params], "%.3f")
print(f"FLOPs: {flops}, Params: {params}")
MMSeg分析Flops和Params
littleyy666的博客
04-25 242
所有data中有数据，同时，input_shape通过默认参数得到，两个不能同时有值，所以将data注释掉。希望通过输入的参数计算。后面可跟参数shape控制输入图片尺寸，例如。这里看一下FlopAnalyzer是如何使用。Flops计算量，params参数量。如何计算某一模块的计算量和参数量呢？在87行左右，由于配置文件在配置了。
计算模型的参数量(Params)和运算量(FLOPs)
AI天才研究院
06-26 4980
衡量模型的好坏，除了任务特定的性能指标(如准确率)，还需要考虑模型的效率，比如模型的参数量和运算量。是指模型的参数个数，描述模型存储所需；通常用衡量，描述模型使用所需。注意是，指浮点运算数量，通常以109)为单位；而不是，指每秒浮点运算次数，后者通常用于衡量硬件的性能指标。对于一个卷积核尺寸为h×w×cin​的卷积层，其输出特征图的尺寸为H×W×cout​cout​×h×w×cin​1H×W×cout​×2×h。
一文讲解thop库计算FLOPs问题
shuaijieer的博客
02-20 6622
计算模型的FLOPs及参数大小FLOPS是处理器性能的衡量指标，是“每秒所执行的浮点运算次数”的缩写。FLOPs是算法复杂度的衡量指标，是“浮点运算次数”的缩写，s代表的是复数。一般使用thop库来计算，GitHub：但官网的Readme中详细写出了是用来计算MACs，而不是FLOPs的MACs（Multiply-Accumulates）和 FLOPs（Floating-Point Operations）都是用来度量深度学习模型的计算量的指标。
Pytorch 中计算Flops和Params
weixin_55406683的博客
03-17 432
统计模型Flops 和 Params
使用thop输出深度学习模型的flops和param
yanzhizhiyang的博客
03-13 265
这里文件的位置我随便写的，使用时换成自己的模型路径就ok。
两个输入计算模型flops
最新发布
qq_47225799的博客
07-03 280
我的模型有两个输入，但是在forward定义时，输入的时候给定的是一个，然后使用x[0]和x[1]分别取出第一和第二个输入。这时会因为参数问题报错。
计算模型FLOPs和参数量（计算工具）
Charles5101的博客
08-08 4008
在pytorch环境下，有两个计算FLOPs和参数量的包thop和ptflops，结果基本是一致的。 thop 参考https://github.com/Lyken17/pytorch-OpCounter 安装方法：pip install thop 使用方法： from torchvision.models import resnet18 from thop import profile model = resnet18() input = torch.randn(1, 3, 224, 224)
使用thop计算放入GPU中的模型的参数量和FLOPs
毛毛的博客
10-09 1720
安装 pip install thop 使用 input = torch.randn(1, 3, 1024).to(device) # 放进和model相同的GPU中 flops, params = profile(model, inputs=(input,), verbose=True) print("%s ------- params: %.2fMB ------- flops: %.2fG" % (model, params / (1000 ** 2), flops / (1000 ** 3)))
使用thop库对yolo等深度学习模型的FLOPS进行计算
沉醉于风中的博客
03-16 2968
据说yolov5原来的FLOPS计算脚本有bug，因此这个大神推荐使用thop库进行计算，代码如下： input = torch.randn(1, 3, 416, 416) flops, params = thop.profile(model, inputs=(input,)) print('flops:', flops / 900000000*2) print('params:', params) 收藏在这里，方便以后需要算yolov5等模型的FLOPS时使用； 本文章来源 知乎pogg：YOLOv5
【25】模型参数量Params与计算量Flops的计算方法
Clichong
01-05 4035
文章目录1. 推导公式1.1 CNN Params1.2 CNN Flops1.3 Linear Params1.4 Linear Flops2. 计算方法2.1 parameters法2.2 thop法2.3 torchstat法2.4 torchsummary法 1. 推导公式 1.1 CNN Params 1.2 CNN Flops 1.3 Linear Params 1.4 Linear Flops 2. 计算方法 2.1 parameters法 通过遍历模型的parameters获取结
FLOPs、FLOPS、Params的含义及PyTorch中的计算方法
weixin_44966641的博客
09-04 2万+
FLOPs、FLOPS、Params的含义及PyTorch中的计算方法 含义解释 FLOPS：注意全大写，是floating point operations per second的缩写（S表second秒），表示每秒浮点运算次数，理解为计算速度。是一个衡量硬件性能的指标。 FLOPs：注意s小写，是floating point operations的缩写（s表复数），表示浮点运算数，理解为计算量。可以用来衡量算法/模型的复杂度。 Params：没有固定的名称，大小写均可，表示模型的参数量，也是用来衡量算法
THOP: 统计 PyTorch 模型的 FLOPs 和参数量
热门推荐
yiran103的专栏
07-30 3万+
THOP 是 PyTorch 非常实用的一个第三方库，可以统计模型的 FLOPs 和参数量。使用方法为： from thop import clever_format from thop import profile class YourModule(nn.Module): # your definition def count_your_model(model, x, y): ......
Pytorch查看模型参数量和计算量
mytzs123的博客
06-29 1万+
1、首先声明一个网络使用torchsummary可以查看模型的参数，和输入输出尺寸，但不能看FLOPs。 2、pip install thop可以显示总的FLOPs和参数量params，但不能显示每层的结构，有一些警告，但是没关系，因为relu是不参与计算参数量和FLOPs的。 3、pip install torchstat可以显示总的FLOPs和参数量params，也可能显示每层的结构。.........
Python/Pytorch常用函数大汇总（持续更新中）
qq_52353238的博客
02-04 1003
nargs - 应该读取的命令行参数个数，可以是具体的数字，或者是?在上面的代码中，我们先导入了argparse这个包，然后包中的ArgumentParser类生成一个parser对象（好多博客中把这个叫做参数解析器），其中的description描述这个参数解析器是干什么的，当我们在命令行显示帮助信息的时候会看到description描述的信息。而对于tensor的计算操作，默认是要进行计算图的构建的，在这种情况下，可以使用 with torch.no_grad():，强制之后的内容不进行计算图构建。
Pytorch中计算自己模型的FLOPs(profile(),clever_format())
java_pythons的博客
03-27 1345
Pytorch中计算自己模型的FLOPs(profile(),clever_format())
Pytorch模型分析：计算Pytorch模型的FLOPs、模型参数量、MAdd、模型显存占用量
噢啦啦耶的博客
08-05 1万+
参考资料1：https://dotnet.ctolib.com/Swall0w-torchstat.html 参考资料2：https://www.jianshu.com/p/6514b8fb1ada 参考资料3：https://www.jianshu.com/p/cbada26ea29d 参考资料4：https://blog.csdn.net/qq_40329272/article/details/106797617   由于模型分析的需要，除了对比模型在指定任务上的表现外，我们可能还需要评估模型的FLOP
python-基础-计算模型参数总量
SKY_yiyi_9的博客
08-16 8173
计算模型参数总量 方法1：pytorch自带方法，计算模型参数总量 total = sum([param.nelement() for param in model.parameters()]) print(“Number of parameter: %.2fM” % (total/1e6)) 例子： 红色断点放到下一行，并用debug运行。结果为： 方法2：用工具包thop 要安装：pip install thop 在程序里输入： from thop import profile from thop
Pytorch统计模型参数量和计算量
weixin_42096202的博客
01-13 9147
需要使用官方库THOP: PyTorch-OpCounter GitHub 地址：https://github.com/Lyken17/pytorch-OpCounter 1.安装THOP： pip install thop # or pip install --upgrade git+https://github.com/Lyken17/pytorch-OpCounter.git 2.使用方...
CNN模型中卷积的一些计算--参数量，FLOPs
ZSH12345678900的博客
08-06 505
feature map经卷积后的大小： conv: h−f+2ps+1\frac{h-f+2p}{s}+1sh−f+2p​+1 —fff: kernel size; ppp: padding; sss: stride dilated conv: h+2p−(f−1)×ds+1\frac{h+2p-(f-1)×d}{s}+1sh+2p−(f−1)×d​+1 — ddd: dilated pooling: h−fs+1\frac{h-f}{s}+1sh−f​+1 receptive filed计算（从前往后
pytorch计算flops
12-02
以下是使用pytorch计算flops的方法： ```python ...在引用中提到，使用profile算出来的flops需要乘以2，而thop计算出来的flops则不需要乘以2。因此，在使用不同的计算方法时，需要注意其计算规则的差异。
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值