added checkpoint test on cpu

2026-09-10 12:21:57 +08:00 · 2019-07-26 11:51:25 -04:00
parent 51a5cc36e3
commit 2ee8f157ce
1 changed files with 1 additions and 1 deletions
@@ -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()