lightning/pytorch_lightning/accelerators
Carlos Mocholí 81d15c5986
Implement double optimizer closure for hook structure consistency (#10167)
2021-10-29 13:03:04 +00:00
..
__init__.py CI: yesqa (#8564) 2021-08-02 16:05:56 +00:00
accelerator.py Implement double optimizer closure for hook structure consistency (#10167) 2021-10-29 13:03:04 +00:00
cpu.py [2/4] Add DeviceStatsMonitor callback (#9712) 2021-10-13 18:29:36 +00:00
gpu.py Avoid deprecation warning after #9901 (#9951) 2021-10-16 17:36:25 +01:00
ipu.py Isolate optimizer step logic to the `PrecisionPlugin` (#10029) 2021-10-20 15:43:08 +00:00
tpu.py Isolate optimizer step logic to the `PrecisionPlugin` (#10029) 2021-10-20 15:43:08 +00:00