被自己都蠢哭了~~
一直報錯沒有調整過來,以爲是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