0
点赞
收藏
分享

微信扫一扫

[chrome devtools]Sources面板

鲤鱼打个滚 2023-10-17 阅读 30
python

问题:
在这里插入图片描述
解决:
在这里插入图片描述
添加strict=False,允许加载过程中出现不匹配的键。但请注意,仍然需要确保模型中的主要参数能够正确加载,以确保模型的有效性。

model.load_state_dict(state_dict)
# 改为:
model.load_state_dict(state_dict, strict=False)
举报

相关推荐

0 条评论