diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index ceb5e397..8594f731 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -30,7 +30,7 @@ def reduce_distributed_output(output, nb_gpus): return output def dummy(gpu, a): - print(gpu) + print(a) class Trainer(TrainerIO):