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

猜你喜欢

转载自blog.csdn.net/weixin_36411839/article/details/103920040
今日推荐