Skip to content

size mismatch for Prediction.bias Error! #1315

Open
@Chiyan200

Description

@Chiyan200

This code is used
import easyocr
reader = easyocr.Reader(['ta'], gpu=False) # this needs to run only once to load the model into memory
result = reader.readtext('tamTxt.png')
print("result : ",result)

i got this error i run this code using cpu
Traceback (most recent call last):
File "C:\python\test\g.py", line 2, in
reader = easyocr.Reader(['ta'], gpu=False) # this needs to run only once to load the model into memory
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "C:\Users\Python312\Lib\site-packages\easyocr\easyocr.py", line 231, in init
self.recognizer, self.converter = get_recognizer(recog_network, network_params,
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "C:\Python\Python312\Lib\site-packages\easyocr\recognition.py", line 174, in get_recognizer
model.load_state_dict(new_state_dict)
File "C:\Python\Python312\Lib\site-packages\torch\nn\modules\module.py", line 2215, in load_state_dict
raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format(
RuntimeError: Error(s) in loading state_dict for Model:
size mismatch for Prediction.weight: copying a param with shape torch.Size([143, 512]) from checkpoint, the shape in current model is torch.Size([127, 512]).
size mismatch for Prediction.bias: copying a param with shape torch.Size([143]) from checkpoint, the shape in current model is torch.Size([127]).

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions