当前位置：网站首页>目标检测模型大小计算，模型复杂度（参数换算公式）

目标检测模型大小计算，模型复杂度（参数换算公式）

2022-07-17 09:37:00 【纯粹ss】

目标检测模型大小计算，模型复杂度（参数换算公式）

最近在做目标检测的时候需要对比模型大小，因为有的模型输出结果不包括模型大小，只输出参数量如图
在这里插入图片描述

或者这种
在这里插入图片描述
，而这时候保存的每个epoch的pt文件不仅仅是参数权重的大小，而是整个checkpoint，包括优化器等等其他内容。
我因此也是苦恼了好久才找到一个通过参数量来估算模型大小的方法，或者说公式，经过我对比许多模型的计算结果来看是比较准确的，
参数换算公式: 参数量 * 2 / 1024 /1024（一个float16占2个字节， 1024个字节为1k）

原网站

版权声明
本文为[纯粹ss]所创，转载请带上原文链接，感谢
https://blog.csdn.net/qq_45294476/article/details/125824258

边栏推荐

猜你喜欢

随机推荐