lightning/tests/tests_pytorch/strategies
Dan Dale 4f82068bcf
Generalize `Optimizer` validation to accommodate both FSDP 1.x and 2.x (#16733)
Co-authored-by: Jirka Borovec <6035284+Borda@users.noreply.github.com>
Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
2023-03-28 12:32:21 +00:00
..
launchers Merge DDPStrategy and DDPSpawnStrategy in PL (#16809) 2023-02-27 14:43:23 -05:00
scripts copyright Lightning AI team (#16647) 2023-02-06 15:26:51 +01:00
__init__.py
test_common.py Remove DP (#16748) 2023-02-16 05:12:08 +00:00
test_custom_plugin.py copyright Lightning AI team (#16647) 2023-02-06 15:26:51 +01:00
test_ddp.py Call `_cuda_clearCublasWorkspaces` on teardown (#16907) 2023-03-06 17:52:02 +01:00
test_ddp_spawn.py Add `Strategy.on_exception` (#16646) 2023-02-08 14:00:31 +00:00
test_ddp_spawn_strategy.py Replace obsolete `_FakeQueue` in multiprocessing launcher (#16873) 2023-02-28 01:31:18 +00:00
test_ddp_strategy.py Set a fixed stage in the evaluation loops (#17094) 2023-03-27 10:00:03 +02:00
test_ddp_strategy_with_comm_hook.py Merge DDPStrategy and DDPSpawnStrategy in PL (#16809) 2023-02-27 14:43:23 -05:00
test_deepspeed_strategy.py Add check for bf16 in deepspeed inference (#16973) 2023-03-21 21:29:23 +00:00
test_fsdp.py Generalize `Optimizer` validation to accommodate both FSDP 1.x and 2.x (#16733) 2023-03-28 12:32:21 +00:00
test_registry.py Merge DDPStrategy and DDPSpawnStrategy in PL (#16809) 2023-02-27 14:43:23 -05:00
test_single_device_strategy.py Set a fixed stage in the evaluation loops (#17094) 2023-03-27 10:00:03 +02:00
test_xla.py Sequential `CombinedLoader` to flatten the eval and predict loops (#16726) 2023-02-17 17:37:11 +01:00