I don’t know if there is a way to do that but there is load_best_model_at_end=True
. But I’m not quite sure if it is something you want. Maybe see here: Save only best model in Trainer - #5 by sgugger
I don’t know if there is a way to do that but there is load_best_model_at_end=True
. But I’m not quite sure if it is something you want. Maybe see here: Save only best model in Trainer - #5 by sgugger