15 lines
1.1 KiB
Python
15 lines
1.1 KiB
Python
from pytorch_lightning.accelerators.cpu_accelerator import CPUAccelerator
|
|
from pytorch_lightning.accelerators.ddp2_accelerator import DDP2Accelerator
|
|
from pytorch_lightning.accelerators.ddp_accelerator import DDPAccelerator
|
|
from pytorch_lightning.accelerators.ddp_spawn_accelerator import DDPSpawnAccelerator
|
|
from pytorch_lightning.accelerators.ddp_cpu_spawn_accelerator import DDPCPUSpawnAccelerator
|
|
from pytorch_lightning.accelerators.dp_accelerator import DataParallelAccelerator
|
|
from pytorch_lightning.accelerators.gpu_accelerator import GPUAccelerator
|
|
from pytorch_lightning.accelerators.tpu_accelerator import TPUAccelerator
|
|
from pytorch_lightning.accelerators.horovod_accelerator import HorovodAccelerator
|
|
from pytorch_lightning.accelerators.ddp_slurm_accelerator import DDPSLURMAccelerator
|
|
from pytorch_lightning.accelerators.ddp_torchelastic_accelerator import DDPTorchElasticAccelerator
|
|
from pytorch_lightning.accelerators.ddp_cpu_torchelastic_accelerator import DDPCPUTorchElasticAccelerator
|
|
from pytorch_lightning.accelerators.ddp_cpu_slurm_accelerator import DDPCPUSLURMAccelerator
|
|
from pytorch_lightning.accelerators.accelerator import Accelerator
|