1. 碎碎念
搞不懂,論文里面的模型大小(Modelsize)
到底是啥?感覺各有各的說法
- 就單純的網(wǎng)絡(luò)結(jié)構(gòu)的大?。?font color="blue">那就可以說成是:網(wǎng)絡(luò)結(jié)構(gòu)定義好了就是一個模型,可用于訓(xùn)練
- 還是整個權(quán)重(網(wǎng)絡(luò)結(jié)構(gòu)+參數(shù))的大???可以說成:整個權(quán)重文件包含了網(wǎng)絡(luò)結(jié)構(gòu)和參數(shù),那么這個訓(xùn)練好的權(quán)重文件可直接使用,用于嵌入到某些平臺或設(shè)備上就可以使用了,如果沒有訓(xùn)練好的參數(shù),單純就一個網(wǎng)絡(luò)結(jié)構(gòu)那在應(yīng)用中也是沒有作用的
2. 換算公式1(僅計算網(wǎng)絡(luò)結(jié)構(gòu)大小)
目標(biāo)檢測模型大小計算,模型復(fù)雜度(參數(shù)換算公式)
文章來源:http://www.zghlxwxcb.cn/news/detail-754904.html
3. 換算公式2(計算訓(xùn)練后的權(quán)重文件大?。?/h2>
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')
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')
其實這個統(tǒng)計出來的就是在文件夾中能看到的文件大小(也就是右鍵該文件能看到的大小屬性):文章來源地址http://www.zghlxwxcb.cn/news/detail-754904.html

到了這里,關(guān)于【論文】模型大小計算方式(或者叫做權(quán)重大小/體積?)單位是MB的文章就介紹完了。如果您還想了解更多內(nèi)容,請在右上角搜索TOY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章,希望大家以后多多支持TOY模板網(wǎng)!