Sean Naren 3b0e4e0b2b Enable ZeRO tests for CI, fix to/half function calls (#6070) ... * Enable ZeRO optimization, and make sure that the lightning module hook is called when we move to half precision * Added test, update to function		2021-02-21 00:24:44 +00:00
..
__init__.py	Refactor LightningDataParallel (#5670)	2021-01-31 06:08:16 -05:00
base.py	Enable ZeRO tests for CI, fix to/half function calls (#6070)	2021-02-21 00:24:44 +00:00
data_parallel.py	Hardware specific parts of Accelerator Refactoring (#5719)	2021-02-01 08:34:59 -05:00
distributed.py	Refactor LightningDataParallel (#5670)	2021-01-31 06:08:16 -05:00
fairscale.py	PoC: Accelerator refactor (#5743)	2021-02-12 15:48:56 -05:00