fixed correct module on hpc save

2019-07-24 18:20:56 -04:00 · 2019-07-24 18:20:56 -04:00 · a0e2b5ee54
parent 8f0d9af168
commit a0e2b5ee54
1 changed files with 1 additions and 0 deletions
--- a/tests/test_models.py
+++ b/tests/test_models.py
@ -307,6 +307,7 @@ def run_gpu_model_test(trainer_options, model, hparams, on_gpu=True):

    if trainer.use_ddp:
        trainer.model = pretrained_model
+        trainer.optimizers = pretrained_model.configure_optimizers()

    # test HPC loading / saving
    trainer.hpc_save(save_dir, exp)