目标检测模型大小计算，模型复杂度（参数换算公式）
发布时间：2023-04-10 16:00
目标检测模型大小计算，模型复杂度（参数换算公式）
最近在做目标检测的时候需要对比模型大小，因为有的模型输出结果不包括模型大小，只输出参数量如图
或者这种
，而这时候保存的每个epoch的pt文件不仅仅是参数权重的大小，而是整个checkpoint，包括优化器等等其他内容。
我因此也是苦恼了好久才找到一个通过参数量来估算模型大小的方法，或者说公式，经过我对比许多模型的计算结果来看是比较准确的，
参数换算公式: 参数量 * 2 / 1024 /1024（一个float16占2个字节， 1024个字节为1k）