wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-09 11:32:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

added clean slurm save load test

Browse Source
This commit is contained in:
William Falcon
2019-07-26 22:40:07 -04:00
parent 3224365190
commit 61c82611eb
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_models.py
+1 -2
View File
@@ -80,9 +80,8 @@ def test_cpu_slurm_save_load():
trainer.model = LightningTestModel(hparams)
# test HPC loading
trainer.global_step = 20000000
trainer.hpc_load(save_dir, on_gpu=False)
assert trainer.global_step == real_global_step and trainer.global_step != 20000000
assert trainer.global_step == real_global_step and trainer.global_step > 0
# predict with loaded model to make sure answers are the same
trainer.model.eval()