From 516ee9c985495fc23bbf91c4d7d26affb4127e3f Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 24 Jul 2019 17:21:18 -0400 Subject: [PATCH] added multiple outputs to LightningTestModel --- pytorch_lightning/testing_models/lm_test_module.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch_lightning/testing_models/lm_test_module.py b/pytorch_lightning/testing_models/lm_test_module.py index c0843667cd..09c12c4dad 100644 --- a/pytorch_lightning/testing_models/lm_test_module.py +++ b/pytorch_lightning/testing_models/lm_test_module.py @@ -18,7 +18,7 @@ class LightningTestModel(LightningModule): Sample model to show how to define a template """ - def __init__(self, hparams, force_remove_distributed_sampler): + def __init__(self, hparams, force_remove_distributed_sampler=False): """ Pass in parsed HyperOptArgumentParser to the model :param hparams: