Hi all, I have trained a model and saved it, tokenizer as well. During the training I set the load_best_checkpoint_at_end to True and can see the test results, which are good Now I have another file where I load the model and observe results on test data set. I want to be able to do this without t…

I don’t understand the question. With load_best_model_at_end the model loaded at the end of training is the one that had the best performance on your validation set. So when you save that model, you have the best model on this validation set. If it’s crap on another set, it means your validation se…

Loading a model from local with best checkpoint

Beginners

valhalla October 29, 2020, 5:29pm 4

I am not sure if that’s possible with Trainer right now, pinging @sgugger

1 Like

Topic		Replies	Views
Question Regarding trainer arguments:: load_best_model_at_end Beginners	2	1953	April 19, 2021
Do trainer.save_model saves the best model? 🤗Transformers	3	6369	July 3, 2023
Does checkpoint have memory in the case of resume from checkpoint Beginners	0	223	February 28, 2024
Checkpoint vs model weight Beginners	2	4786	October 12, 2020
Differences in prediction from train end to checkpoint Beginners	3	843	September 11, 2023

Loading a model from local with best checkpoint

Related topics