set max_steps and save limit so we have the checkpoint when testing r…

…esuming
axolotl-ai-cloud · Nov 15, 2023 · ac88f7a · ac88f7a
1 parent e70edd9
commit ac88f7a
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tests/e2e/test_resume.py b/tests/e2e/test_resume.py
@@ -59,7 +59,9 @@ def test_resume_qlora(self, temp_dir):
                 "learning_rate": 0.00001,
                 "optimizer": "adamw_torch",
                 "lr_scheduler": "cosine",
-                "save_steps": 10,
+                "save_steps": 30,
+                "save_total_limit": 5,
+                "max_steps": 40,
             }
         )
         if is_torch_bf16_gpu_available():
@@ -90,4 +92,4 @@ def test_resume_qlora(self, temp_dir):
         )
         pattern = r"first_step\s+(\d+)"
         first_steps = int(re.findall(pattern, res.stdout)[0])
-        assert first_steps == 11
+        assert first_steps == 31