winglian commited on
Commit
3fbde76
1 Parent(s): f6060a6

fix save_steps so it doesn't get duplicated (#567)

Browse files
Files changed (1) hide show
  1. src/axolotl/utils/trainer.py +0 -1
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=(