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

fixed correct module on hpc save

Browse Source
This commit is contained in:
William Falcon committed 2019-07-24 18:11:29 -04:00
1 parent 10330f1991
commit 2e0fde7da7
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
@@ -277,7 +277,7 @@ def run_gpu_model_test(trainer_options, model, hparams, on_gpu=True):
# test HPC loading / saving
trainer.hpc_save(save_dir, exp)
trainer.hpc_load(save_dir, on_gpu=True)
trainer.hpc_load(save_dir, on_gpu=on_gpu)
clear_save_dir()