diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index 71267c34..887648c1 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -82,6 +82,7 @@ class Trainer(TrainerIO): self.print_nan_grads = print_nan_grads self.data_parallel_device_ids = None self.world_size = 1 + print('-'*100) # gpus come in as a string. # if gpus = -1 then use all available devices