Loading model from checkpoint after error in training

The checkpoint should be saved in a directory that will allow you to go model = XXXModel.from_pretrained(that_directory).

6 Likes