fixed correct module on hpc save

2019-07-24 18:11:29 -04:00 · 2019-07-24 18:11:29 -04:00 · 2e0fde7da7
parent 10330f1991
commit 2e0fde7da7
1 changed files with 1 additions and 1 deletions
--- a/tests/test_models.py
+++ b/tests/test_models.py
@ -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()