mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-10-28 13:20:31 +08:00
ddp pickle
This commit is contained in:
1 parent
d2989f76e6
commit
2181ad1bc7
1 file changed
+1
@@ -134,6 +134,7 @@ class TrainerDataLoadingMixin(ABC):
|
||||
'ddp': self.num_nodes * self.num_processes,
|
||||
'ddp2': self.num_nodes,
|
||||
}
|
||||
import pdb; pdb.set_trace()
|
||||
sampler = DistributedSampler(
|
||||
dataloader.dataset,
|
||||
num_replicas=world_size.get(self.distributed_backend, 0),
|
||||
|
||||
Reference in new issue
Block a user