fix: suspected eval vram increased usage

This commit is contained in:
NanoCode012
2025-06-23 18:44:03 +07:00
parent 0494359c6c
commit a65dbe779f

View File

@@ -215,10 +215,6 @@ class HFCausalTrainerBuilder(TrainerBuilderBase):
self.cfg.auto_find_batch_size
)
training_arguments_kwargs["eval_accumulation_steps"] = (
self.cfg.gradient_accumulation_steps
)
training_arguments_kwargs["load_best_model_at_end"] = (
(
self.cfg.load_best_model_at_end is not False