超级会员免费看
本文介绍了如何使用thop库来计算PyTorch模型的参数数量。首先，通过pip安装thop，如果常规安装失败，提供了一个备用的安装方法。接着，文章引用了CSDN上的资源，展示了利用thop统计张量元素数量以及计算模型参数量的方法。
1. 安装 thop
1.1 常规安装
pip install thop
1.2 若上述安装方式错误，可以参考以下方式：
pip install thop-i http://pypi.douban.com/simple --trusted-host pypi.douban.com
参考：https://editor.csdn.net/md/?articleId=1064660472. 使用thop计算参数量和计算量
from nets.yolo4 import YoloBody ## 导入模型
from thop import profile ## 导入thop模块
model = YoloBody(3,20)
input = torch.randn(1, 3, 416, 416)
flops, params = profile(model, inputs=(input,))
print('flops',flops) ## 打印计算量
print('params',params) ## 打印参数量
原文链接：https://blog.csdn.net/weixin_44239195/article/details/110390705
统计张量元素个数：
订阅专栏 解锁全文
扫一扫
3991
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
