reduce number of steps

axolotl-ai-cloud · Oct 30, 2024 · 4e64549 · 4e64549
1 parent 4924072
commit 4e64549
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/e2e/multigpu/test_llama.py b/tests/e2e/multigpu/test_llama.py
@@ -631,7 +631,7 @@ def test_8bit_lora_ds_zero3(self, temp_dir):
                     },
                 ],
                 "num_epochs": 1,
-                "max_steps": 50,
+                "max_steps": 15,
                 "micro_batch_size": 4,
                 "gradient_accumulation_steps": 4,
                 "output_dir": temp_dir,