lightning/tests/tests_pytorch/strategies
Carlos Mocholí bdd9b12c5a
Refactor `loop.setup_data` with utility functions (#16918)
2023-03-09 00:04:42 +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 Call `_cuda_clearCublasWorkspaces` on teardown (#16907) 2023-03-06 17:52:02 +01: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 Update torchmetrics requirement from <0.10.1,>=0.7.0 to >=0.7.0,<0.11.1 in /requirements (#15904) 2023-03-02 23:35:31 +00:00
test_fsdp.py PL: Test PyTorch 2.0 pre-release on CPU and CUDA (#16764) 2023-02-28 18:38:40 +01:00
test_registry.py Merge DDPStrategy and DDPSpawnStrategy in PL (#16809) 2023-02-27 14:43:23 -05:00
test_single_device_strategy.py Refactor `loop.setup_data` with utility functions (#16918) 2023-03-09 00:04:42 +00:00
test_xla.py Sequential `CombinedLoader` to flatten the eval and predict loops (#16726) 2023-02-17 17:37:11 +01:00