當前位置:編程學習大全網 - 網站源碼 - 改變mobilenetv2網絡後沒辦法使用預訓練權重怎麽半

改變mobilenetv2網絡後沒辦法使用預訓練權重怎麽半

題主是否想詢問“改變mobilenetv2網絡後沒辦法使用預訓練權重怎麽辦”?<br>1、首先對於微調後的網絡模型可以將load_state_dict()函數中的參數strict=False。

2、其次當strict = True時,要求加載的權重文件中的參數與模型中的完全匹配,設置strict = False時,即可使用權重文件中與模型匹配層的數據,其余的默認初始化。

3、最後當模型更改後不能如上述情況加載預訓練權重時,可以考慮獲取原模型的權重加載到新模型上。

  • 上一篇:WLAN轉WIFI
  • 下一篇:壹次性搞懂 Promise、async await 與 Generator
  • copyright 2024編程學習大全網