deep-text-recognition-benchmark icon indicating copy to clipboard operation
deep-text-recognition-benchmark copied to clipboard

does pretrained model not include space character?

Open akbarwijayaa opened this issue 3 years ago • 2 comments

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

akbarwijayaa avatar Jan 12 '22 11:01 akbarwijayaa

@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

AhmadZobairSurosh avatar Mar 15 '22 13:03 AhmadZobairSurosh

@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?

ftmasadi avatar Dec 11 '22 11:12 ftmasadi