I have saved my model using the code torch.save(the_model.state_dict(), PATH) after training
while loading, I am confused. can someone explain this code
the_model = TheModelClass(*args, **kwargs) the_model.load_state_dict(torch.load(PATH))