PyTorch加載模型model.load_state_dict()問題,Unexpected key(s) in state_dict: "module.features..,Expected .

希望將訓練好的模型加載到新的網絡上。如上面題目所描述的,PyTorch在加載之前保存的模型參數的時候,遇到了問題。     Unexpected key(s) in state_dict: "module.features. ...".,Expected ".features....". 直接原因是key值名字不對應。     表明了加載過程中,期望獲得的key值爲feature...,而不是mo
相關文章
相關標籤/搜索