diff --git a/tests/test_models.py b/tests/test_models.py index 2dc05489de..fd2140092f 100644 --- a/tests/test_models.py +++ b/tests/test_models.py @@ -45,7 +45,7 @@ def test_dp_output_reduce(): def test_cpu_slurm_managed(): """ - SLURM checkpointing works + Verify model save/load/checkpoint on CPU :return: """ hparams = get_hparams()