wassname/pytorch-lightning
1
0 Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-10-29 13:30:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

added checkpoint test on cpu

Browse Source
This commit is contained in:
William Falcon committed 2019-07-26 11:51:25 -04:00
1 parent 51a5cc36e3
commit 2ee8f157ce
1 file changed
+1 -1
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_models.py
+1 -1
View File
@@ -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()