clovaai / deep-text-recognition-benchmark

Text recognition (optical character recognition) with deep learning methods, ICCV 2019
Apache License 2.0
3.77k stars 1.11k forks source link

does pretrained model not include space character? #325

Open akbarwijayaa opened 2 years ago

akbarwijayaa commented 2 years ago

hello everyone, i use TPS-ResNet-BiLSTM-Attn-case-sensitive.pth pretrained model, is this pretrained model include space character?

AhmadZobairSurosh commented 2 years ago

@akbarwijayaa Hi Did you get this error too while using : TPS-ResNet-BiLSTM-Attn-case-sensitive.pth """ raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format( RuntimeError: Error(s) in loading state_dict for DataParallel: size mismatch for module.Prediction.attention_cell.rnn.weight_ih: copying a param with shape torch.Size([1024, 352]) from checkpoint, the shape in current model is torch.Size([1024, 294]). size mismatch for module.Prediction.generator.weight: copying a param with shape torch.Size([96, 256]) from checkpoint, the shape in current model is torch.Size([38, 256]). size mismatch for module.Prediction.generator.bias: copying a param with shape torch.Size([96]) from checkpoint, the shape in current model is torch.Size([38])."""

if yes, can you please share with me the solution?

Thanks in advance

ftmasadi commented 2 years ago

@akbarwijayaa Hi Did you get this error too while using : TPS-ResNet-BiLSTM-Attn-case-sensitive.pth """ raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format( RuntimeError: Error(s) in loading state_dict for DataParallel: size mismatch for module.Prediction.attention_cell.rnn.weight_ih: copying a param with shape torch.Size([1024, 352]) from checkpoint, the shape in current model is torch.Size([1024, 294]). size mismatch for module.Prediction.generator.weight: copying a param with shape torch.Size([96, 256]) from checkpoint, the shape in current model is torch.Size([38, 256]). size mismatch for module.Prediction.generator.bias: copying a param with shape torch.Size([96]) from checkpoint, the shape in current model is torch.Size([38])."""

if yes, can you please share with me the solution?

Thanks in advance

Please help me if you find a solution for this issue?