unexpected key “model.conv1.weight” in state_dict

作者分享了在加载模型参数时遇到的意外key错误“model.conv1.weight”,并详细解释了问题的根源在于载入模型参数的方法有误。正确的做法是直接使用model.load_state_dict(statedict),而非model_reid.model.load_state_dict(statedict)。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

被自己都蠢哭了~~

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

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值