Lines Matching refs:DistributedDataParallel
42 from torch.nn.parallel import DistributedDataParallel
1595 cpu_model = DistributedDataParallel(
1606 gpu_model = DistributedDataParallel(
1670 cpu_model = DistributedDataParallel(
1679 gpu_model = DistributedDataParallel(
1707 model = DistributedDataParallel(
1750 model = DistributedDataParallel(
1801 DistributedDataParallel._set_params_and_buffers_to_ignore_for_model(
1805 DistributedDataParallel(
1871 ddp_withload = DistributedDataParallel(
1875 ddp_withoutload = DistributedDataParallel(
1949 ddp_model = DistributedDataParallel(
1975 cpu_model = DistributedDataParallel(
1990 gpu_model = DistributedDataParallel(
2028 model = DistributedDataParallel(
2055 model = DistributedDataParallel(
2098 model = DistributedDataParallel(
2127 ddp_model = DistributedDataParallel(