joblib加载模型报KeyError: 0错误

项目场景:

通过joblib库加载模型,报错代码如下

classifier = joblib.load(classifier_path)

问题描述

报错如下

D:\ANACONDA\envs\pytorch-gpu\python.exe “D:/GUI (5)/GUI/test.py”
Traceback (most recent call last):
File “D:/GUI (5)/GUI/test.py”, line 4, in
x = joblib.load(code_path + “/Model_Weight/Bag_of_Features/details_files”)
File “D:\ANACONDA\envs\pytorch-gpu\lib\site-packages\joblib\numpy_pickle.py”, line 587, in load
obj = _unpickle(fobj, filename, mmap_mode)
File “D:\ANACONDA\envs\pytorch-gpu\lib\site-packages\joblib\numpy_pickle.py”, line 506, in _unpickle
obj = unpickler.load()
File “D:\ANACONDA\envs\pytorch-gpu\lib\pickle.py”, line 1088, in load
dispatchkey[0]
KeyError: 0

原因分析:

更新一下joblib版本就可以了,太蠢了!这是我之前的版本
在这里插入图片描述


解决方案:

更新成1.3.1就好了,蠢!
在这里插入图片描述

  • 3
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
根据提供的引用内容,报错信息是"KeyError: 'state_dict_best'"。根据引用\[1\]和引用\[3\]的代码片段,可以看出在加载模型时出现了键错误。这可能是由于加载的权重文件中缺少了名为"state_dict_best"的键。为了解决这个问题,可以尝试以下几种方法: 1. 检查权重文件:确保权重文件中存在名为"state_dict_best"的键。如果不存在,可能需要重新下载或获取正确的权重文件。 2. 修改加载代码:根据引用\[2\]的代码片段,可以尝试修改加载模型的代码,使用正确的键来加载权重文件。例如,使用`torch.load("densenet169_rnn_fold_1_model_best_f1.pth.tar")\['state_dict'\]`来替换`torch.load("densenet169_rnn_fold_1_model_best_f1.pth.tar")\['state_dict_best'\]`。 3. 使用strict=False参数:根据引用\[3\]的代码片段,可以尝试在加载模型时使用`strict=False`参数,以允许加载过程中出现未知键的情况。例如,使用`model.load_state_dict(torch.load(weights, map_location=device)\['model'\], strict=False)`来替换`model.load_state_dict(torch.load(weights, map_location=device)\['model'\], strict=True)`。 请根据具体情况选择适合的解决方法来解决报错问题。 #### 引用[.reference_title] - *1* *2* *3* [Missing key(s) in state_dict | unexpected_keys | 模型加载 [torch.load] 报错【解决方案】](https://blog.csdn.net/sinat_28442665/article/details/120451762)[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^control_2,239^v3^insert_chatgpt"}} ] [.reference_item] [ .reference_list ]

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值