unexpected key “model.conv1.weight” in state_dict

程序员文章站 2022-05-27 10:32:41

...

被自己都蠢哭了～～

一直报错没有调整过来，以为是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

pytorch加载模型报错Missing key(s) in state_dict: “classifier00.0.fc.weight“, “classifier00.0.fc.bias“