Lines Matching refs:DistributedDataParallel
41 from torch.nn.parallel import DistributedDataParallel
301 model = DistributedDataParallel(
987 ddp_model = DistributedDataParallel(
994 ddp_model = DistributedDataParallel(
1002 ddp_model = DistributedDataParallel(model, process_group=process_group)
1008 ddp_model = DistributedDataParallel(
1018 ddp_model = DistributedDataParallel(
1080 model = DistributedDataParallel(
1156 DistributedDataParallel(
1204 model = DistributedDataParallel(
1211 model = DistributedDataParallel(
1342 model = DistributedDataParallel(
1395 model = DistributedDataParallel(
1511 ddp = DistributedDataParallel(
1537 ddp = DistributedDataParallel(
1617 m_ddp = DistributedDataParallel(
1731 m_ddp = DistributedDataParallel(
1739 m_ddp = DistributedDataParallel(
1752 gpu_model = DistributedDataParallel(
2069 m = torch.nn.parallel.DistributedDataParallel(
2120 lstm_ddp = DistributedDataParallel(
2193 ddp_model = DistributedDataParallel(
2334 ddp = DistributedDataParallel(
3499 ddp_model = DistributedDataParallel(model)