Searched defs:DistributedDataParallel (Results 1 – 6 of 6) sorted by relevance
46 def register_ddp(self, ddp: DistributedDataParallel) -> None:69 def register_ddp(self, ddp_inst: DistributedDataParallel):
176 ddp: DistributedDataParallel,338 ddp: DistributedDataParallel,
1211 NCCL < 2.9.61221 NCCL >= 2.9.6
172 DistributedDataParallel section in Implementation
326 .. _amp-custom-examples:
326 class DistributedDataParallel(Module, Joinable): class