From 93a2eccc1fe87acf9824f3f60c1d78186c2f1335 Mon Sep 17 00:00:00 2001 From: NanoCode012 Date: Fri, 14 Feb 2025 20:11:31 +0700 Subject: [PATCH] fix: update default max_steps --- src/axolotl/core/trainer_builder.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/axolotl/core/trainer_builder.py b/src/axolotl/core/trainer_builder.py index fbe6b982f7..ab9fc39ccc 100755 --- a/src/axolotl/core/trainer_builder.py +++ b/src/axolotl/core/trainer_builder.py @@ -327,7 +327,7 @@ def _set_base_training_args(self, total_num_steps) -> dict[str, Any]: self.cfg.save_total_limit if self.cfg.save_total_limit else 4 ) - training_args_kwargs["max_steps"] = self.cfg.max_steps or total_num_steps + training_args_kwargs["max_steps"] = self.cfg.max_steps or total_num_steps or -1 # max_length is not used in CausalTrainer if self.cfg.reward_model or self.cfg.rl: