From 72e96f255d86274bf0f6503167024eeb75ed3d1f Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Wed, 13 Sep 2023 20:40:33 -0400 Subject: [PATCH] fix save_steps so it doesn't get duplicated (#567) --- src/axolotl/utils/trainer.py | 1 - 1 file changed, 1 deletion(-) diff --git a/src/axolotl/utils/trainer.py b/src/axolotl/utils/trainer.py index 946505c222..555737f761 100644 --- a/src/axolotl/utils/trainer.py +++ b/src/axolotl/utils/trainer.py @@ -640,7 +640,6 @@ def setup_trainer(cfg, train_dataset, eval_dataset, model, tokenizer, total_num_ eval_accumulation_steps=cfg.gradient_accumulation_steps, num_train_epochs=cfg.num_epochs, learning_rate=cfg.learning_rate, - save_steps=cfg.save_steps, output_dir=cfg.output_dir, save_total_limit=cfg.save_total_limit if cfg.save_total_limit else 4, load_best_model_at_end=(