@JuanFMontesinos Thank you I’ll give it a shot.
@ecdrid Because the “new” state_dict is of different size, for example the original fc1 is a [5,10] tensor while the one in the state_dict is a [2,10] and it would give the following error:
RuntimeError: Error(s) in loading state_dict for Net:
	size mismatch for fc1.weight: copying a param with shape torch.Size([2, 10]) from checkpoint, the shape in current model is torch.Size([5, 10]).
	size mismatch for fc2.weight: copying a param with shape torch.Size([2, 2]) from checkpoint, the shape in current model is torch.Size([2, 5]).