lightning/pytorch_lightning/accelerators
Sean Naren b7c2e0a80e
Trainer only references accelerator (#6039)
* Trainer only references accelerator where it can

* Move teardown to the trainer, as it is reponsible for the accelerator
2021-02-17 21:41:18 +01:00
..
__init__.py PoC: Accelerator refactor (#5743) 2021-02-12 15:48:56 -05:00
accelerator.py Trainer only references accelerator (#6039) 2021-02-17 21:41:18 +01:00
cpu.py Hardware specific parts of Accelerator Refactoring (#5719) 2021-02-01 08:34:59 -05:00
gpu.py move device-specific teardown logic from training loop to accelerator (#5973) 2021-02-15 17:38:03 -05:00
tpu.py PoC: Accelerator refactor (#5743) 2021-02-12 15:48:56 -05:00