From 6e2bf991f0f2491901ce82c043797cd9a5a89081 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 24 Jul 2019 18:21:22 -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 63d38b4ddc..1c6693b5e4 100644 --- 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()