mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-10-28 13:20:31 +08:00
ddp fix
This commit is contained in:
1 parent
a24c88ab08
commit
63addd091c
1 file changed
-1
@@ -378,7 +378,6 @@ class TrainerDDPMixin(ABC):
|
||||
:param model:
|
||||
:return:
|
||||
"""
|
||||
import pdb; pdb.set_trace()
|
||||
if self.proc_rank == 0:
|
||||
path = os.path.join(self.default_root_dir, '__temp_weight_ddp_end.ckpt')
|
||||
self.save_checkpoint(path)
|
||||
|
||||
Reference in new issue
Block a user