Add logic to handle saving the last checkpoint regardless of val chec…

…k interval (NVIDIA#4905) Signed-off-by: SeanNaren <snarenthiran@nvidia.com> Signed-off-by: SeanNaren <snarenthiran@nvidia.com> Signed-off-by: Matvei Novikov <mattyson.so@gmail.com>
jubick1337 · Oct 3, 2022 · e317d9a · e317d9a
1 parent 4fc4d38
commit e317d9a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/core/test_exp_manager.py b/tests/core/test_exp_manager.py
@@ -571,4 +571,4 @@ class CustomLoop(TrainingEpochLoop):
         loop.trainer = trainer
         trainer.fit_loop.epoch_loop = loop
         with pytest.warns(UserWarning, match="Detected custom epoch loop"):
-            exp_manager(trainer, {"explicit_log_dir": str(tmp_path)})
+            exp_manager(trainer, {"explicit_log_dir": str(tmp_path)})