问题:
解决:
添加strict=False,允许加载过程中出现不匹配的键。但请注意,仍然需要确保模型中的主要参数能够正确加载,以确保模型的有效性。
model.load_state_dict(state_dict)# 改为:model.load_state_dict(state_dict, strict=False)
© 版权声明
文章版权归作者所有,未经允许请勿转载。
THE END
问题:
解决:
添加strict=False,允许加载过程中出现不匹配的键。但请注意,仍然需要确保模型中的主要参数能够正确加载,以确保模型的有效性。
model.load_state_dict(state_dict)# 改为:model.load_state_dict(state_dict, strict=False)