0


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

目录

1. 碎碎念

搞不懂,论文里面的

  1. 模型大小(Modelsize)

到底是啥?感觉各有各的说法

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

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

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

在这里插入图片描述

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

  1. import os
  2. defgetfilesize_single(filepath):
  3. byte_size = os.path.getsize(filepath)# 单位是Byte (1KB = 1024Byte, 1MB = 1024KB)# print(byte_size)
  4. pt_size =round(byte_size /1024/1024,3)# 换算后,单位是MBreturn pt_size
  5. if __name__ =='__main__':
  6. weightpath =r'G:\pycharmprojects\autodl_trainruns\yolov5s-baseline-911split\weights\best.pt'print(f'{getfilesize_single(weightpath)} MB')

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


本文转载自: https://blog.csdn.net/LWD19981223/article/details/129940694
版权归原作者 孟孟单单 所有, 如有侵权,请联系我们删除。

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

还没有评论