題主是否想詢問“改變mobilenetv2網絡後沒辦法使用預訓練權重怎麽辦”?<br>1、首先對於微調後的網絡模型可以將load_state_dict()函數中的參數strict=False。
2、其次當strict = True時,要求加載的權重文件中的參數與模型中的完全匹配,設置strict = False時,即可使用權重文件中與模型匹配層的數據,其余的默認初始化。
3、最後當模型更改後不能如上述情況加載預訓練權重時,可以考慮獲取原模型的權重加載到新模型上。
上一篇:WLAN轉WIFI下一篇:壹次性搞懂 Promise、async await 與 Generator