forked from nudt_dsp/netrans
15 lines
645 B
Python
15 lines
645 B
Python
from .parallel_apply import parallel_apply
|
|
from .replicate import replicate
|
|
from .data_parallel import DataParallel, data_parallel
|
|
from .scatter_gather import scatter, gather
|
|
from .distributed import DistributedDataParallel
|
|
|
|
__all__ = ['replicate', 'scatter', 'parallel_apply', 'gather', 'data_parallel',
|
|
'DataParallel', 'DistributedDataParallel']
|
|
|
|
def DistributedDataParallelCPU(*args, **kwargs):
|
|
import warnings
|
|
warnings.warn("torch.nn.parallel.DistributedDataParallelCPU is deprecated, "
|
|
"please use torch.nn.parallel.DistributedDataParallel instead.")
|
|
return DistributedDataParallel(*args, **kwargs)
|