lightning/pytorch_lightning/accelerators
William Falcon 2b255a3df4
ref: enable custom clusters (1/n) (#4048)
* enable cluster plugins

* enable cluster plugins + test backend choices

* enable cluster plugins + test backend choices

* enable cluster plugins + test backend choices

* enable cluster plugins + test backend choices

* enable cluster plugins + test backend choices

* enable cluster plugins + test backend choices
2020-10-10 08:09:29 -04:00
..
__init__.py test selecting the correct backend. temp backends while slurm and TE are decoupled (#3848) 2020-10-04 15:44:50 -04:00
accelerator_connector.py ref: enable custom clusters (1/n) (#4048) 2020-10-10 08:09:29 -04:00
base_accelerator.py ref: enable custom clusters (1/n) (#4048) 2020-10-10 08:09:29 -04:00
cpu_backend.py clean and organize fit (#3938) 2020-10-07 11:04:10 -04:00
ddp2_backend.py clean and organize fit (#3938) 2020-10-07 11:04:10 -04:00
ddp_backend.py clean and organize fit (#3938) 2020-10-07 11:04:10 -04:00
ddp_cpu_slurm_backend.py clean and organize fit (#3938) 2020-10-07 11:04:10 -04:00
ddp_cpu_spawn_backend.py clean and organize fit (#3938) 2020-10-07 11:04:10 -04:00
ddp_cpu_torchelastic_backend.py clean and organize fit (#3938) 2020-10-07 11:04:10 -04:00
ddp_slurm_backend.py clean and organize fit (#3938) 2020-10-07 11:04:10 -04:00
ddp_spawn_backend.py clean and organize fit (#3938) 2020-10-07 11:04:10 -04:00
ddp_torchelastic_backend.py clean and organize fit (#3938) 2020-10-07 11:04:10 -04:00
dp_backend.py clean and organize fit (#3938) 2020-10-07 11:04:10 -04:00
gpu_backend.py clean and organize fit (#3938) 2020-10-07 11:04:10 -04:00
horovod_backend.py added tests for the training epoch end (#3967) 2020-10-07 22:27:36 -04:00
tpu_backend.py clean and organize fit (#3938) 2020-10-07 11:04:10 -04:00