fix save_steps so it doesn't get duplicated (#567)
Browse files
src/axolotl/utils/trainer.py
CHANGED
@@ -640,7 +640,6 @@ def setup_trainer(cfg, train_dataset, eval_dataset, model, tokenizer, total_num_
|
|
640 |
eval_accumulation_steps=cfg.gradient_accumulation_steps,
|
641 |
num_train_epochs=cfg.num_epochs,
|
642 |
learning_rate=cfg.learning_rate,
|
643 |
-
save_steps=cfg.save_steps,
|
644 |
output_dir=cfg.output_dir,
|
645 |
save_total_limit=cfg.save_total_limit if cfg.save_total_limit else 4,
|
646 |
load_best_model_at_end=(
|
|
|
640 |
eval_accumulation_steps=cfg.gradient_accumulation_steps,
|
641 |
num_train_epochs=cfg.num_epochs,
|
642 |
learning_rate=cfg.learning_rate,
|
|
|
643 |
output_dir=cfg.output_dir,
|
644 |
save_total_limit=cfg.save_total_limit if cfg.save_total_limit else 4,
|
645 |
load_best_model_at_end=(
|