lightning/tests/tests_lite/strategies
Adrian Wälchli 05dbf48ad0
Activation checkpointing in FSDP without boilerplate (#15826)
* initial
* input type
* checkpointing
* fsdp in pl
* all_close

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-12-06 15:45:33 +00:00
..
launchers Revert new Hydra launch behavior (#15737) 2022-11-21 20:19:13 +00:00
__init__.py Standalone Lite: Launchers (#14555) 2022-09-12 14:15:42 +00:00
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 Activation checkpointing in FSDP without boilerplate (#15826) 2022-12-06 15:45:33 +00:00
test_fsdp_integration.py Activation checkpointing in FSDP without boilerplate (#15826) 2022-12-06 15:45:33 +00: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