pytorch加载的模型踩坑之Error(s) in loading state_dict

根据引用\[1\]中的报错信息，可以看出加载模型时出现了RuntimeError，提示缺少了一些键值。根据引用\[2\]的方法1，可以通过设置`strict=False`来加载模型，但这可能导致一些参数加载不进来，进而影响推理结果的准确性。另外，根据引用\[3\]的方法，可以尝试在加载模型时指定`map_location='cpu'`来解决报错问题。具体的加载模型代码如下： ```python ckpt = torch.load(model_dir, map_location='cpu') net.load_state_dict(ckpt\['model_state_dict'\]) ``` 这样可以尝试解决`RuntimeError`报错问题。 #### 引用[.reference_title] - *1* *2* *3* [PyTorch加载模型时报错RuntimeError: Error(s) in loading state_dict for *****: Missing key(s) in state...](https://blog.csdn.net/mj412828668/article/details/130014232)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^insert_down28v1,239^v3^insert_chatgpt"}} ] [.reference_item] [ .reference_list ]