RuntimeError: Error(s) in loading state_dict for Model: size mismatch for Prediction.weight: copying a param with shape torch.Size([174, 512]) from checkpoint, the shape in current model is torch.Size([185, 512]). size mismatch for Prediction.bias: copying a param with shape torch.Size([174]) from checkpoint, the shape in current model is torch.Size([185]).
这取决于你是否要尝试手动注入字符。 [在Python中加载yaml,Windows操作系统]: 在我的情况下,预期大小为97,但实际上为99。当我尝试在easyocr.py中打印字符时,看起来符号€变成了€,它应该仍然是€。 问题在于使用yaml加载时字符发生了变化。因此,为了调试,我在easyocr.py文件中直接通过分配我的字符来更改了手动输入。 self.character = "my character same in my custom_example.yaml" 在你的情况下,预期大小应该是174,但实际上是185。
9条答案
按热度按时间avwztpqn1#
RuntimeError: Error(s) in loading state_dict for Model: size mismatch for Prediction.weight: copying a param with shape torch.Size([174, 512]) from checkpoint, the shape in current model is torch.Size([185, 512]). size mismatch for Prediction.bias: copying a param with shape torch.Size([174]) from checkpoint, the shape in current model is torch.Size([185]).
gmol16392#
我遵循了你的建议,但仍然得到了相同的错误 #487
@falhafizh
wlp8pajw3#
@kimlia545
How can I use my custom model? please help me
ozxc1zmp4#
请按照以下指令操作:https://github.com/JaidedAI/EasyOCR/blob/master/custom_model.md
mf98qq945#
我通过了这一部分并应用了更改,但我又遇到了这个错误。我认为这个错误的原因是我的字符与默认字符之间的差异。我需要在代码中做些什么吗?
dba5bblo6#
这取决于你是否要尝试手动注入字符。
[在Python中加载yaml,Windows操作系统]:
在我的情况下,预期大小为97,但实际上为99。当我尝试在easyocr.py中打印字符时,看起来符号€变成了€,它应该仍然是€。
问题在于使用yaml加载时字符发生了变化。因此,为了调试,我在easyocr.py文件中直接通过分配我的字符来更改了手动输入。
self.character = "my character same in my custom_example.yaml"
在你的情况下,预期大小应该是174,但实际上是185。
h9a6wy2h7#
尝试在Google Colab中加载。
cmssoen28#
非常感谢您的帮助,手动注入字符是否足够在easyocr.py中输入这行代码(self.character = "my character same in my custom_example.yaml")?
cyej8jka9#
对于我的情况,是的。
我的备选方案是尝试在Google Colab中运行easyocr并加载我的模型。它正常运行(不需要更改任何内容)。