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

ddp fix

This commit is contained in:
William Falcon 2020-04-27 08:28:39 -04:00
parent a24c88ab08
commit 63addd091c
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
pytorch_lightning/trainer/distrib_data_parallel.py
View File

@ -378,7 +378,6 @@ class TrainerDDPMixin(ABC):
:param model: :param model:
:return: :return:
""" """
import pdb; pdb.set_trace()
if self.proc_rank == 0: if self.proc_rank == 0:
path = os.path.join(self.default_root_dir, '__temp_weight_ddp_end.ckpt') path = os.path.join(self.default_root_dir, '__temp_weight_ddp_end.ckpt')
self.save_checkpoint(path) self.save_checkpoint(path)