You should’t get that error using new_models.load_state_dict(torch.load('./cifar_net.pth')), because this is the right way to load a model, not new_models=torch.load('./cifar_net.pth'), beacuse this doesn’t load the weights to the model, but only loading the dict on the new_model variable