目标检测模型大小计算,模型复杂度(参数换算公式)

本文介绍了如何计算目标检测模型的大小,特别是当模型输出只包含参数量时。作者分享了一个估算模型大小的公式:参数量*2/1024/1024,基于float16占2个字节进行换算。这个方法经过验证,对于估算模型文件大小较为准确,有助于在模型优化和资源管理中进行比较。
摘要由CSDN通过智能技术生成

目标检测模型大小计算,模型复杂度(参数换算公式)

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

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

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值