【论文】模型大小计算方式(或者叫做权重大小/体积?)单位是MB

1. 碎碎念

搞不懂,论文里面的模型大小(Modelsize)到底是啥?感觉各有各的说法

  1. 就单纯的网络结构的大小?那就可以说成是:网络结构定义好了就是一个模型,可用于训练
  2. 还是整个权重(网络结构+参数)的大小?可以说成:整个权重文件包含了网络结构和参数,那么这个训练好的权重文件可直接使用,用于嵌入到某些平台或设备上就可以使用了,如果没有训练好的参数,单纯就一个网络结构那在应用中也是没有作用的

2. 换算公式1(仅计算网络结构大小)

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

在这里插入图片描述

3. 换算公式2(计算训练后的权重文件大小)

import os

def getfilesize_single(filepath):
    byte_size = os.path.getsize(filepath)  # 单位是Byte (1KB = 1024Byte, 1MB = 1024KB)
    # print(byte_size)
    pt_size = round(byte_size / 1024 / 1024, 3)  # 换算后,单位是MB
    return pt_size


if __name__ == '__main__':
    weightpath = r'G:\pycharmprojects\autodl_trainruns\yolov5s-baseline-911split\weights\best.pt'
    print(f'{getfilesize_single(weightpath)} MB')

其实这个统计出来的就是在文件夹中能看到的文件大小(也就是右键该文件能看到的大小属性):

  • 7
    点赞
  • 35
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

孟孟单单

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值