lightning/tests/tests_lite/strategies
Adrian Wälchli 657bfc586a
Fix device placement when setting up FSDP model in Lite (#15822)
* fix
* debug test
* simplify

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-28 04:05:48 +01:00
..
launchers Revert new Hydra launch behavior (#15737) 2022-11-21 20:19:13 +00:00
__init__.py
test_ddp.py Efficient gradient accumulation in LightningLite (#14966) 2022-10-19 19:55:12 +00:00
test_deepspeed.py Support individual setup of model and optimizer in Lite (#15185) 2022-11-11 14:36:59 +01:00
test_deepspeed_integration.py [Lite] precision_plugin -> precision (#15001) 2022-10-10 15:00:32 +00:00
test_dp.py Standalone Lite: DataParallel Strategy (#14681) 2022-09-14 19:27:53 -04:00
test_fairscale.py Support individual setup of model and optimizer in Lite (#15185) 2022-11-11 14:36:59 +01:00
test_fairscale_integration.py Grep for potential errors in standalone tests (#15341) 2022-11-05 04:29:38 +01:00
test_fsdp.py FSDP (native) support for LightningLite (#14967) 2022-11-21 13:58:37 +00:00
test_fsdp_integration.py Fix device placement when setting up FSDP model in Lite (#15822) 2022-11-28 04:05:48 +01:00
test_registry.py FSDP (native) support for LightningLite (#14967) 2022-11-21 13:58:37 +00:00
test_single_device.py Standalone Lite: Strategy base classes and registry (#14662) 2022-09-14 09:15:21 -04:00
test_xla.py Lite: setting extras & fix CI (#15192) 2022-10-19 19:05:23 +00:00