Pytorch保存模型的兩種方式:
1 模型結(jié)構(gòu)和模型參數(shù)都保存下來
優(yōu)點(diǎn):不需要預(yù)初始化模型,直接加載,就可以獲得模型和參數(shù)入蛆,只要把數(shù)據(jù)放進(jìn)去就可以進(jìn)行測(cè)試
缺點(diǎn):每個(gè)模型存儲(chǔ)的文件比較大屋摔,一般的會(huì)有幾個(gè)G简十。
2 只存儲(chǔ)模型的參數(shù)檬某,不存儲(chǔ)模型結(jié)構(gòu)
優(yōu)點(diǎn):存儲(chǔ)文件較小
前提: 需要預(yù)初始化模型