lightning/tests/tests_fabric/strategies
Bas Krahmer 420eb6f248
Added configurable strict loading for Fabric strategies (#17645)
Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
Co-authored-by: bas <bas.krahmer@talentflyxpert.com>
Co-authored-by: Jirka Borovec <6035284+Borda@users.noreply.github.com>
2023-06-06 18:26:13 -04:00
..
launchers update list of fist party packages (#16859) 2023-03-03 16:55:48 +00:00
__init__.py Rename LightningLite to Fabric (#16244) 2023-01-04 10:57:18 -05:00
test_ddp.py Address feedback for `Fabric.init_module()` (4/4) (#17607) 2023-06-03 02:07:02 +00:00
test_deepspeed.py ruff: autofix PT (#17541) 2023-05-04 11:50:39 -04:00
test_deepspeed_integration.py Refactor run-method-style Fabric tests (#17669) 2023-05-21 09:04:01 -04:00
test_dp.py add & apply flake8-simplify (#17386) 2023-04-24 21:57:08 +00:00
test_fsdp.py [FSDP] utility to apply optimizer during backward (#17710) 2023-06-06 21:41:26 +02:00
test_fsdp_integration.py Added configurable strict loading for Fabric strategies (#17645) 2023-06-06 18:26:13 -04:00
test_registry.py [TPU] Rename classes to use XLA instead of TPU (#17383) 2023-04-28 12:36:22 -04:00
test_single_device.py Address feedback for `Fabric.init_module()` (4/4) (#17607) 2023-06-03 02:07:02 +00:00
test_strategy.py Added configurable strict loading for Fabric strategies (#17645) 2023-06-06 18:26:13 -04:00
test_xla.py [TPU] For XLA Strategy, added function arg to control `broadcast_master_param()` (#17522) 2023-05-05 17:57:24 +00:00