Oh the documentation is outdated, you shouldn’t use your model from the checkpoint directory anymore, as long as the checkpoint is in the output_dir
, the Trainer
will use it if you do trainer.train(resume_from_checkpoint=True)
.
You can also pass the folder to your exact checkpoint instead of True
.