2021-12-22 20:23:30 +00:00
|
|
|
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
|
2022-01-03 12:41:05 +00:00
|
|
|
from pytorch_lightning.strategies.ipu import IPUStrategy # noqa: F401
|
2021-12-22 20:23:30 +00:00
|
|
|
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
|
2021-12-23 14:01:23 +00:00
|
|
|
from pytorch_lightning.strategies.strategy import Strategy # noqa: F401
|
2021-12-22 20:23:30 +00:00
|
|
|
from pytorch_lightning.strategies.tpu_spawn import TPUSpawnStrategy # noqa: F401
|