added checkpoint test on cpu

2019-07-26 11:51:25 -04:00 · 2019-07-26 11:51:25 -04:00 · 2ee8f157ce
parent 51a5cc36e3
commit 2ee8f157ce
1 changed files with 1 additions and 1 deletions
--- 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()