From 2ee8f157ce851eaf5ae9597397aae22d57429a13 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Fri, 26 Jul 2019 11:51:25 -0400 Subject: [PATCH] added checkpoint test on cpu --- tests/test_models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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()