diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index 3a03f4cc..a79af0d4 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -449,6 +449,9 @@ class Trainer(TrainerIO): except KeyError as e: node_rank = 0 + print('-'*100) + print(gpu_nb) + print('-'*100) # recover original exp before went into process # init in write mode only on proc 0 self.experiment.debug = self.proc_rank > 0