Searched refs:_get_data_parallel_params (Results 1 – 3 of 3) sorted by relevance
104 for param in ddp_inst._get_data_parallel_params(ddp_inst.module):
2255 def _get_data_parallel_params(module, named_params=False): member in DistributedDataParallel
4774 dp_params = torch.nn.parallel.DistributedDataParallel._get_data_parallel_params(4784 …dp_params = torch.nn.parallel.DistributedDataParallel._get_data_parallel_params(model, named_param…