reduce number of steps

2024-10-30 14:51:10 -04:00
parent ef60e3e851
commit 920ea77bdf
1 changed files with 1 additions and 1 deletions
--- a/tests/e2e/multigpu/test_llama.py
+++ b/tests/e2e/multigpu/test_llama.py
@@ -632,7 +632,7 @@ class TestMultiGPULlama:
                    },
                ],
                "num_epochs": 1,
-                "max_steps": 50,
+                "max_steps": 15,
                "micro_batch_size": 4,
                "gradient_accumulation_steps": 4,
                "output_dir": temp_dir,