| 1234567891011121314151617181920 |
- # mypy: allow-untyped-defs
- from typing_extensions import deprecated
- from .parallel_apply import parallel_apply
- from .replicate import replicate
- from .data_parallel import DataParallel, data_parallel
- from .scatter_gather import gather, scatter
- from .distributed import DistributedDataParallel
- __all__ = ['replicate', 'scatter', 'parallel_apply', 'gather', 'data_parallel',
- 'DataParallel', 'DistributedDataParallel']
- @deprecated(
- "`torch.nn.parallel.DistributedDataParallelCPU` is deprecated, "
- "please use `torch.nn.parallel.DistributedDataParallel` instead.",
- category=FutureWarning,
- )
- def DistributedDataParallelCPU(*args, **kwargs):
- return DistributedDataParallel(*args, **kwargs)
|
