THOP: PyTorch-OpCounter

How to install

pip install thop (now continously intergrated on Github actions)

pip install --upgrade git+https://github.com/Lyken17/pytorch-OpCounter.git

How to use

Basic usage

from torchvision.models import resnet50
from thop import profile
model = resnet50()
input = torch.randn(1, 3, 224, 224)
macs, params = profile(model, inputs=(input, ))

Define the rule for 3rd party module.

class YourModule(nn.Module):# your definition
def count_your_model(model, x, y):# your rule hereinput = torch.randn(1, 3, 224, 224)
macs, params = profile(model, inputs=(input, ), custom_ops={YourModule: count_your_model})

Improve the output readability

Call thop.clever_format to give a better format of the output.
```
from thop import clever_format
macs, params = clever_format([macs, params], "%.3f")
```

Results of Recent Models

The implementation are adapted from torchvision. Following results can be obtained using benchmark/evaluate_famous_models.py.

Model	Params(M)	MACs(G)
alexnet	61.10	0.77
vgg11	132.86	7.74
vgg11_bn	132.87	7.77
vgg13	133.05	11.44
vgg13_bn	133.05	11.49
vgg16	138.36	15.61
vgg16_bn	138.37	15.66
vgg19	143.67	19.77
vgg19_bn	143.68	19.83
resnet18	11.69	1.82
resnet34	21.80	3.68
resnet50	25.56	4.14
resnet101	44.55	7.87
resnet152	60.19	11.61
wide_resnet101_2	126.89	22.84
wide_resnet50_2	68.88	11.46

Model	Params(M)	MACs(G)
resnext50_32x4d	25.03	4.29
resnext101_32x8d	88.79	16.54
densenet121	7.98	2.90
densenet161	28.68	7.85
densenet169	14.15	3.44
densenet201	20.01	4.39
squeezenet1_0	1.25	0.82
squeezenet1_1	1.24	0.35
mnasnet0_5	2.22	0.14
mnasnet0_75	3.17	0.24
mnasnet1_0	4.38	0.34
mnasnet1_3	6.28	0.53
mobilenet_v2	3.50	0.33
shufflenet_v2_x0_5	1.37	0.05
shufflenet_v2_x1_0	2.28	0.15
shufflenet_v2_x1_5	3.50	0.31
shufflenet_v2_x2_0	7.39	0.60
inception_v3	27.16	5.75

词库加载错误:未能找到文件“E:\highferrum_mysql\Configuration\Dict_Stopwords.txt”。

上一篇：CDGA|数字化转型成功的关键要素

下一篇：Linux系统中内核态、用户态和零拷贝技术解析

热门资讯

unity教程｜｜Unity调... 记录下，个人之前做一个项目的时候碰到一个需求，安卓端的app中需要做冷更...

自然语言处理（四）： Part... 目录 1. What is Part of Speech (POS)? 词性是什么 2. Infor...

著名央视主持人在甘孜救人，现场... 转自齐鲁晚报 5月28日四川甘孜州文广旅局局长刘洪转发的一条视频引发关注视频中，央视主持人...

女孩趴在藏獒身上，结果发生了让...

京郊首个高速停车区主题文化公园... 文、图/赵向红 5月30日，京雄高速（进京方向）房山南停车区内热闹非凡，由“中铁高速”和中国收藏家协...

酒店、机场、高铁项目最爱：铝合...

【内蒙古】网友露营大型翻车现场...

姑苏首店，玩转假期！当青石板路遇见首店热潮当园林雅韵碰撞潮流文化姑苏区正用首发经济书写古城商业的破圈秘籍近期，苏...

缺失的第一个正整数：给定一个未... 给定一个未排序的整数数组，找出其中未出现的最小正整数。 (本文获得CSDN质量评分...

圣祥乳业惊艳第六届北京牛奶文化... 5月29日至31日，由北京市奶业协会主办的第六届北京牛奶文化节在北京中关村移动智能服务创新园（上地・...

糖醋鱼去哪学？超实用课程，手把...

水饺培训能学多久？短期课程，让...

原创拌... #图文打卡计划#大家好呀！今天咱们来聊聊一个让无数厨房新手头疼的问题——为啥自己拌的凉菜总是水汪汪的...

剩米饭大变身！10 分钟做出黄... 家里的剩米饭总在冰箱里 “吃灰”？别浪费！只需 10 分钟，就能将平淡无奇的剩米饭变身金黄酥脆、香气...

原创空... 端午了，别就知道吃粽子了！我们吃点新鲜的，艾草戚风蛋糕、奶油粽宝三明治还有脆皮粽子......总有...

如何为孩子挑选健康的零食？儿童的饮食习惯是他们健康成长的重要基石，然而，如何为孩子挑选健康的零食却让家长感到困惑。根据数据显示...

男人会做饭生活更浪漫_家常菜_...

一口熨斗糕，品尝重庆的童年甜蜜... 在重庆的美食记忆里，熨斗糕宛如一颗甜蜜的糖果，带着童年的纯真与欢乐，让人回味无穷。它那外酥里嫩的口感...

慷慢油莎豆奶为你的健康加一餐

爱情、火锅和踅面夫妻创业比蜜... 王丽他生在延安市黄龙县，她诞于韩城市；他总是怀念乡味铜火锅，她一直爱吃家乡的踅面。两人一合计，走!...

[pytorch]thop计算模型算力和参数量

THOP: PyTorch-OpCounter

How to install

How to use

Results of Recent Models

相关内容

热门资讯