From a0e2b5ee54af289c050ce996e5ca5d1f81177706 Mon Sep 17 00:00:00 2001
From: William Falcon <falc@fb.com>
Date: Wed, 24 Jul 2019 18:20:56 -0400
Subject: [PATCH] fixed correct module on hpc save

---
 tests/test_models.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tests/test_models.py b/tests/test_models.py
index 3a364aea27..63d38b4ddc 100644
--- 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)