I see, thank you for your answer.
I have a follow up question, let say my model is model = models.resnet101(pretrained = True) and then I finetune it, and then want to save it and load it again.
Do you suggest to save and load it like:
torch.save(the_model.state_dict(), PATH)
the_model.load_state_dict(torch.load(PATH))
or you have other suggestion?
in link 3
it says to do
the_model = TheModelClass(*args, **kwargs)
first, but i dont get why and what it means by TheModelClass