RuntimeError: Error(s) in loading state_dict for Discriminator: size mismatch for conv2.weight: copying a param with shape torch.Size([5, 2048, 4, 4]) from checkpoint, the shape in current model is torch.Size([5, 2048, 2, 2]).

[Somyarani13]

RuntimeError: Error(s) in loading state_dict for Discriminator:
size mismatch for conv2.weight: copying a param with shape torch.Size([5, 2048, 4, 4]) from checkpoint, the shape in current model is torch.Size([5, 2048, 2, 2]).

I was running your prototype for the star-Gan and got the above error. Can you help me resolve it. For the generator the pretrained weights was loaded successfully but for discriminator I was getting this error.