I am using trainer to finetune llava model. I want to save the best model during training based on the specific metric, however I found it loads the model at the end of training instead of the best one during the training. Here’s my relevant hyperparameters: trainer.train() trainer.evaluate(eval_da…

Load_best_model_at_end doesn't work?

fabikru March 25, 2025, 12:46pm 2

I have the same issue. Did anybody find a solution for this yet?

1 Like

Topic		Replies	Views
Unexpected behavior of load_best_model_at_end in Trainer (or am I doing it wrong?) 🤗Transformers	2	75	March 25, 2025
Question Regarding trainer arguments:: load_best_model_at_end Beginners	2	1962	April 19, 2021
Save only best model in Trainer 🤗Transformers	31	86158	June 25, 2024
Hyperparameter optimization and load_best_model_at_end 🤗Transformers	2	892	June 22, 2023
Trainer "load_best_model_at_end" doesn't load the best model Intermediate	0	2592	February 21, 2023