added multiple outputs to LightningTestModel

2019-07-24 17:19:31 -04:00 · 2019-07-24 17:19:31 -04:00 · 0aa91c7fdc
parent 3521e87286
commit 0aa91c7fdc
1 changed files with 3 additions and 1 deletions
--- a/tests/test_models.py
+++ b/tests/test_models.py
@ -190,7 +190,9 @@ def test_amp_gpu_ddp():

    os.environ['MASTER_PORT'] = str(np.random.randint(12000, 19000, 1)[0])

-    model, hparams = get_model()
+    hparams = get_hparams()
+    model = LightningTestModel(hparams)
+
    trainer_options = dict(
        progress_bar=True,
        max_nb_epochs=1,