lightning/pytorch_lightning/overrides/__init__.py

3 lines
185 B
Python

from pytorch_lightning.overrides.data_parallel import LightningParallelModule # noqa: F401
from pytorch_lightning.overrides.distributed import LightningDistributedModule # noqa: F401