Open akbarwijayaa opened 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
@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?
hello everyone, i use TPS-ResNet-BiLSTM-Attn-case-sensitive.pth pretrained model, is this pretrained model include space character?