unexpected key “model.conv1.weight” in state_dict

被自己都蠢哭了~~

一直報錯沒有調整過來,以爲是key的問題,後來發現是自己導入key的方式有問題,週末要好好研究一下這個 state_dict,以免之後再次出現這種超級低級的錯誤!!

之前的問題是
unexpected key “model.conv1.weight” in state_dict

其實是我載入模型參數時,我使用的是

model_reid.model.load_state_dict(statedict)

使用方法有誤,應該使用

model_reid.load_state_dict(statedict)

這樣就會直接去掉model

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章