lightning/pytorch_lightning/accelerators
Carlos Mocholí 74eb6cc7e9
Clean `cuda.empty_cache` usage (#8199)
2021-06-30 13:04:24 +02:00
..
__init__.py IPU Integration 5/5 (#7867) 2021-06-11 15:07:04 +00:00
accelerator.py Refactor unnecessary `else` / `elif` when `if` block has a `return` statement (#8156) 2021-06-28 15:27:41 +05:30
cpu.py release 1.3.0 (#7404) 2021-05-06 15:05:35 -04:00
gpu.py Clean `cuda.empty_cache` usage (#8199) 2021-06-30 13:04:24 +02:00
ipu.py IPU Integration 5/5 (#7867) 2021-06-11 15:07:04 +00:00
tpu.py refactor accelerator teardown -> training type plugin teardown (#7579) 2021-05-22 13:19:24 -07:00