diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index 04b18a148d..909d38df73 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -472,7 +472,7 @@ class Trainer(TrainerIO): self.__run_pretrain_routine(model) - def ddp_train(self, gpu_nb, model, d): + def ddp_train(self, gpu_nb, model): """ Entry point into a DP thread :param gpu_nb: