diff --git a/pytorch_lightning/testing_models/lm_test_module.py b/pytorch_lightning/testing_models/lm_test_module.py index c0843667..09c12c4d 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: