【bug】RuntimeError: Error(s) in loading state_dict for LightweightEdge: Unexpected key(s) in state_di

【bug】RuntimeError: Error(s) in loading state_dict for LightweightEdge: Unexpected key(s) in state_dict

环境

modelscope             1.9.1
torch                  1.13.1+cpu

问题详情

在使用读光-文字识别-轻量化端侧识别模型-中英-通用领域 · 模型库 (modelscope.cn) 模型时弹出这个错误，代码如下

from modelscope.pipelines import pipeline
from modelscope.utils.constant import Tasks
import cv2

ocr_recognition = pipeline(Tasks.ocr_recognition, model='damo/cv_LightweightEdge_ocr-recognitoin-general_damo', model_revision='v2.4.1')

### 使用url
img_url = 'http://duguang-labelling.oss-cn-shanghai.aliyuncs.com/mass_img_tmp_20220922/ocr_recognition.jpg'
result = ocr_recognition(img_url)
print(result)

错误详细信息，注意self.recognizer.load_state_dict(model_dict) 所在文件位置。

Traceback (most recent call last):
  File "C:\Users\xxx\miniconda3\envs\tf115\lib\site-packages\modelscope\utils\registry.py", line 210, in build_from_cfg
    return obj_cls._instantiate(**args)
  File "C:\Users\xxx\miniconda3\envs\tf115\lib\site-packages\modelscope\models\base\base_model.py", line 67, in _instantiate
    return cls(**kwargs)
  File "C:\Users\xxx\miniconda3\envs\tf115\lib\site-packages\modelscope\models\cv\ocr_recognition\model.py", line 104, in __init__
    self.recognizer.load_state_dict(model_dict)
  File "C:\Users\xxxx\miniconda3\envs\tf115\lib\site-packages\torch\nn\modules\module.py", line 1672, in load_state_dict
    self.__class__.__name__, "\n\t".join(error_msgs)))
RuntimeError: Error(s) in loading state_dict for LightweightEdge:
        Unexpected key(s) in state_dict: "head.weight", "head.bias", "our_nas_model.first_conv.0.weight",

错误原因

加载使用模型时和训练模型时的环境不一致，模型是GPU训练与保存的，但我使用的是CPU推理，所以会出现异常。

解决方法

给load_state_dict 的strict参数赋值False
例如，我使用的环境，load_state_dict 位于\modelscope\models\cv\ocr_recognition\model.py ，自行找到对应文件，添加参数后保存文件，例如下图：

如果是cpu环境弹出错误，也可以尝试转到GPU环境

参考

加载模型出现 RuntimeError: Error(s) in loading state_dict for Model: Missing key(s) in state_dict-CSDN博客