Case Sensitive Option Not Working while running demo.py

[abdur75648]

I tried both the commands
py demo.py --Transformation TPS --FeatureExtraction ResNet --SequenceModeling BiLSTM --Prediction Attn --image_folder demo_image/ --saved_model TPS-ResNet-BiLSTM-Attn-case-sensitive.pth & py demo.py --sensitive --Transformation TPS --FeatureExtraction ResNet --SequenceModeling BiLSTM --Prediction Attn --image_folder demo_image/ --saved_model TPS-ResNet-BiLSTM-Attn.pth

but getting this Error:
> 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]).

Please tell if I'm mistakening

[lezwon]

Facing the same issue.

[elvinyhlee]

try this pretrained model instead TPS-ResNet-BiLSTM-Attn-case-sensitive.pth

[rafisoeharto]

try TPS-ResNet-BiLSTM-Attn-case-sensitive.pth , with --sensitive

[ezzaimsoufiane]

Facing the same issue, even with -- sensitive option

[MuntahaShams]

@ezzaimsoufiane have u solved ur issue?

[abdur75648]

Saved model weights are wrong, please try train one yourself from scratch