计算神经网络参数量Params、计算量FLOPs（亲测有效的3种方法）
创始人
2024-05-30 00:18:26
0次

1.stat（cpu统计）

pip install torchstat

from torchstat import statstat(model, (3, 32, 32)) #统计模型的参数量和FLOPs，（3,32,32）是输入图像的size

结果：

 问题：当网络中有自定义参数时，就很有可能漏掉那部分参数对应的统计量；stat好像不支持双输入。

2.summary网络结构对应参数(cuda上面统计)

pip install torchsummary

from torchsummary import summarysummary(model,input_size=(3,32,32))

问题：当网络中有自定义参数时，就很有可能漏掉那部分参数。
结果：

3.统计flops和参数量

pip install thop

from thop import profiledummy_input = torch.randn(1, 3, 32, 32)#.to(device)flops, params = profile(model, (dummy_input,))print('FLOPs: ', flops, 'params: ', params)print('FLOPs: %.2f M, params: %.2f M' % (flops / 1000000.0, params / 1000000.0))

问题：当网络中有自定义参数时，flops和params就很有可能漏掉那部分参数。

参考链接：

6种方法计算神经网络参数量Params、计算量FLOPs、Macs简单代码_python transformer计算量flops代码_工科女博的博客-CSDN博客

网络模型的参数量和FLOPs的计算 Pytorch_网络flop_清纯世纪的博客-CSDN博客

https://zhuanlan.zhihu.com/p/541165764

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

上一篇：3月7日,30秒知全网,精选7个热点

下一篇：蓝桥杯真题31日冲刺 |第一天

相关内容

热门资讯

有没有主角是植物的修仙小说？多... 有没有主角是植物的修仙小说？多多益善。落花时节又逢君，花妖的侧重爱情的极品超能少年
痞子蔡《第一次亲密接触》经典话... 痞子蔡《第一次亲密接触》经典话语我在你香烟上写下自己的名字...这不是补肾的方法，这样的做法也是错误...
鸟兽和木匠读后感 鸟兽和木匠读后感鸟兽和木匠读后感 《鸟兽与木匠》读后感 星期天，我读了一本书，叫《鸟兽与木...
高富帅的具体标准是什么？ 高富帅的具体标准是什么？高富帅没有固定标准随着时代不同估计高会变更高富会更多才算是富帅，不同人心中的...
求，嫁给一个死太监的全文 求，嫁给一个死太监的全文密码:82cm
获奖过的讲文明，懂礼貌的手抄报... 获奖过的讲文明，懂礼貌的手抄报有哪些很多你可以去网上搜一定会有你需要的
孟母三迁原文及翻译 孟母三迁原文及翻译孟母三迁原文及翻译如下：【原文】邹孟轲母，号孟母。其舍近墓。孟子之少时，嬉游为墓间...
好看技术流的网游小说 好看技术流的网游小说网游之盗版神话网游之纵横天下网游之双手剑等等的！！！这基本不错给分啊！！！！
女孩于海24小时一元一分正规麻... 加V【ab120590】【hf420624】【mj120590】红中癞子、跑得快，等等，加不上微信就...
关注麻将24小时红中麻将群@... 微【ab120590】 【mj120590】【hf420624】等风也等你。喜欢打麻将的兄弟姐妹们、...
揭秘一元一分红中麻将群202... 微【ab120590】 【mj120590】【hf420624】（广东一元一分红中癞子爆炸码麻将群）...
玩家必看正规一块红中麻将群全面... 1.进群方式《ab120590》或者《mj120590》《hf420624》--QQ(4434063...
时下最流行资讯一块一分24小时... 一元一分麻将群加群主微【ab120590】【hf420624】 【mj120590】等风也等你。喜欢...
到哪里找盘点十大一元一分麻将群... 微【ab120590】 【mj120590】【hf420624】（广东一元一分红中癞子爆炸码麻将群）...
《西瓜视频》24小时不熄火跑的... 群主微信：【ab120590】 【mj120590】【hf420624】没有三缺一的无奈，手机上的麻...
哪里寻找一元一分红中麻将群20... 1.亮点：一元红中麻将微信“群”—ab120590—hf420624—mj120590—客服Q443...
推荐正规红中麻将跑的快群202... 微【ab120590】 【mj120590】【hf420624】等风也等你。喜欢打麻将的兄弟姐妹们、...
参观一元红中麻将群2024已更... 微【ab120590】 【mj120590】【hf420624】等风也等你。喜欢打麻将的兄弟姐妹们、...
西瓜视频上下分正规红中麻将群@... 加V【ab120590】【hf420624】【mj120590】红中癞子、跑得快，等等，加不上微信就...
我来教大家广东24小时在线一元... 认证群主微信微【ab120590】 【mj120590】【hf420624】（一元俩元红中麻将）（跑...