15 lines
1.1 KiB
Python
15 lines
1.1 KiB
Python
from pytorch_lightning.strategies.ddp import DDPStrategy # noqa: F401
|
|
from pytorch_lightning.strategies.ddp2 import DDP2Strategy # noqa: F401
|
|
from pytorch_lightning.strategies.ddp_spawn import DDPSpawnStrategy # noqa: F401
|
|
from pytorch_lightning.strategies.deepspeed import DeepSpeedStrategy # noqa: F401
|
|
from pytorch_lightning.strategies.dp import DataParallelStrategy # noqa: F401
|
|
from pytorch_lightning.strategies.fully_sharded import DDPFullyShardedStrategy # noqa: F401
|
|
from pytorch_lightning.strategies.horovod import HorovodStrategy # noqa: F401
|
|
from pytorch_lightning.strategies.parallel import ParallelStrategy # noqa: F401
|
|
from pytorch_lightning.strategies.sharded import DDPShardedStrategy # noqa: F401
|
|
from pytorch_lightning.strategies.sharded_spawn import DDPSpawnShardedStrategy # noqa: F401
|
|
from pytorch_lightning.strategies.single_device import SingleDeviceStrategy # noqa: F401
|
|
from pytorch_lightning.strategies.single_tpu import SingleTPUStrategy # noqa: F401
|
|
from pytorch_lightning.strategies.tpu_spawn import TPUSpawnStrategy # noqa: F401
|
|
from pytorch_lightning.strategies.training_type_plugin import Strategy # noqa: F401
|