Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

removed print lines

This commit is contained in:
William Falcon 2019-07-18 16:37:48 -04:00
parent 59d60eaf18
commit 53a1a6d462
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pytorch_lightning/models/trainer.py
View File

@ -372,7 +372,6 @@ class Trainer(TrainerIO):
# when using multi-node or DDP within a node start each module in a separate process # when using multi-node or DDP within a node start each module in a separate process
if self.use_ddp: if self.use_ddp:
print('using ddp')
# must copy only the meta of the exp so it survives pickle/unpickle when going to new process # must copy only the meta of the exp so it survives pickle/unpickle when going to new process
self.experiment = self.experiment.get_meta_copy() self.experiment = self.experiment.get_meta_copy()
task = int(os.environ['SLURM_LOCALID']) task = int(os.environ['SLURM_LOCALID'])
@ -382,7 +381,6 @@ class Trainer(TrainerIO):
# 1 gpu or dp option triggers training using DP module # 1 gpu or dp option triggers training using DP module
# easier to avoid NCCL issues # easier to avoid NCCL issues
elif self.use_dp: elif self.use_dp:
print('using dp')
self.dp_train(model) self.dp_train(model)
# ON CPU # ON CPU