fixed correct module on hpc save

2019-07-24 18:21:22 -04:00 · 2019-07-24 18:21:22 -04:00 · 6e2bf991f0
parent a0e2b5ee54
commit 6e2bf991f0
1 changed files with 1 additions and 0 deletions
--- a/tests/test_models.py
+++ b/tests/test_models.py
@ -306,6 +306,7 @@ def run_gpu_model_test(trainer_options, model, hparams, on_gpu=True):
    run_prediction(model.test_dataloader, pretrained_model)

    if trainer.use_ddp:
+        # on hpc this would work fine... but need to hack it for the purpose of the test
        trainer.model = pretrained_model
        trainer.optimizers = pretrained_model.configure_optimizers()