malllabiisc/InteractE

Why "Error(s) in loading state_dict for InteractE:" ?

Dionysusssss opened this issue · 0 comments

When I tried to run the saved model on test, It show:
RuntimeError: Error(s) in loading state_dict for InteractE:
size mismatch for conv_filt: copying a param with shape torch.Size([96, 1, 11, 11]) from checkpoint, the shape in current model is torch.Size([96, 1, 9, 9]).
size mismatch for bn0.weight: copying a param with shape torch.Size([4]) from checkpoint, the shape in current model is torch.Size([1]).
size mismatch for bn0.bias: copying a param with shape torch.Size([4]) from checkpoint, the shape in current model is torch.Size([1]).
size mismatch for bn0.running_mean: copying a param with shape torch.Size([4]) from checkpoint, the shape in current model is torch.Size([1]).
size mismatch for bn0.running_var: copying a param with shape torch.Size([4]) from checkpoint, the shape in current model is torch.Size([1]).
size mismatch for bn1.weight: copying a param with shape torch.Size([384]) from checkpoint, the shape in current model is torch.Size([96]).
size mismatch for bn1.bias: copying a param with shape torch.Size([384]) from checkpoint, the shape in current model is torch.Size([96]).
size mismatch for bn1.running_mean: copying a param with shape torch.Size([384]) from checkpoint, the shape in current model is torch.Size([96]).
size mismatch for bn1.running_var: copying a param with shape torch.Size([384]) from checkpoint, the shape in current model is torch.Size([96]).
size mismatch for fc.weight: copying a param with shape torch.Size([200, 153600]) from checkpoint, the shape in current model is torch.Size([200, 38400]).