Skip to content

Commit

Permalink
Merge pull request axolotl-ai-cloud#2 from AblateIt/teknium1-revert-e…
Browse files Browse the repository at this point in the history
…val-epochs

Revert eval epochs commit
  • Loading branch information
prateeky2806 committed Aug 15, 2023
2 parents 0a29c6b + d5859a6 commit 03dd3e9
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/axolotl/utils/trainer.py
Original file line number Diff line number Diff line change
Expand Up @@ -199,7 +199,7 @@ def setup_trainer(cfg, train_dataset, eval_dataset, model, tokenizer):
eval_accumulation_steps=cfg.gradient_accumulation_steps,
num_train_epochs=cfg.num_epochs,
learning_rate=cfg.learning_rate,
evaluation_strategy="no" if cfg.val_set_size == 0 else "epoch" if cfg.eval_steps < 1 else "steps",
evaluation_strategy="steps" if cfg.val_set_size > 0 else "no",
save_strategy="steps" if cfg.save_steps else "epoch",
eval_steps=cfg.eval_steps if cfg.val_set_size > 0 else None,
save_steps=cfg.save_steps,
Expand Down

0 comments on commit 03dd3e9

Please sign in to comment.