lightning/pytorch_lightning/pt_overrides/override_data_parallel.py

13 lines
566 B
Python

"""
.. warning:: `override_data_parallel` module has been renamed to `data_parallel` since v0.6.0.
The deprecated module name will be removed in v0.8.0.
"""
from pytorch_lightning.utilities import rank_zero_warn
rank_zero_warn("`override_data_parallel` module has been renamed to `data_parallel` since v0.6.0."
" The deprecated module name will be removed in v0.8.0.", DeprecationWarning)
from pytorch_lightning.overrides.data_parallel import ( # noqa: F402
get_a_var, parallel_apply, LightningDataParallel, LightningDistributedDataParallel)