Do trainer.save_model saves the best model?

As @mihai said, it saves the model currently inside the Trainer. If you call it after Trainer.train(), since load_best_model_at_end will have reloaded the best model, it will save the best model.

3 Likes