After training a version of the model, how to reload it to continue training?

[hjb25286]

After training a version of the model, how to reload it to continue training?

[sunwei925]

Please directly load the model path (model.load_state_dict(torch.load('model path'))) after define the model in the train.py file.