From 3ea25a44db49a86381b332f62374d712b84cdab3 Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Wed, 13 Sep 2023 20:30:18 -0400 Subject: [PATCH] fix save_steps so it doesn't get duplicated --- 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 946505c22..555737f76 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=(