RuntimeError: Error(s) in loading state_dict for VSGCNN:

[viettiepx2]

I run test mode and have errors:
RuntimeError: Error(s) in loading state_dict for VSGCNN:
size mismatch for conv_stage_1.VSGCNN_stage_1_conv_1.weight: copying a param with shape torch.Size([128, 32, 3, 3]) from checkpoint, the shape in current model is torch.Size([32, 32, 3, 3]).
...... and more
RuntimeError: Error(s) in loading state_dict for VSGCNN:
Missing key(s) in state_dict: "conv_stage_1.VSGCNN_stage_1_conv_0.weight",
...... and more
Maybe model pretrained has error.
Thanks!

[yukkwan-yuan]

I have the same problem, can anyone solve it? Thank you

[AbirathR]

Yes same problem..has anyone found a fix?