Revert "checkpoint model on first step callback (#2906)"

This reverts commit 10ba1622f7.
2025-07-15 15:01:12 -04:00
parent 10ba1622f7
commit 6f6d917a99
146 changed files with 9 additions and 419 deletions
--- a/tests/e2e/multigpu/test_eval.py
+++ b/tests/e2e/multigpu/test_eval.py
@@ -67,7 +67,6 @@ class TestMultiGPUEval:
                "logging_steps": 1,
                "weight_decay": 0.0,
                "use_tensorboard": True,
-                "save_first_step": False,
            }
        )

@@ -139,7 +138,6 @@ class TestMultiGPUEval:
                "logging_steps": 1,
                "weight_decay": 0.0,
                "use_tensorboard": True,
-                "save_first_step": False,
            }
        )