ChatGPT is wrong as it suggested to change the model, which doesn’t make sense.
You can just restore the model directly, load the state_dict, and continue training.
You’ve already trained the model before and after restoring its state_dict just continue training with a new Dataset, DataLoader, etc.