lightning/tests/tests_pytorch/strategies
Carlos Mocholí c9452df005
Run XLA's dataloader validation per dataloader (#16775)
2023-02-16 15:09:23 +01:00
..
launchers copyright Lightning AI team (#16647) 2023-02-06 15:26:51 +01: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 Set `find_unused_parameters=False` as the default (#16611) 2023-02-06 16:51:21 +01:00
test_ddp_spawn.py Add `Strategy.on_exception` (#16646) 2023-02-08 14:00:31 +00:00
test_ddp_spawn_strategy.py Set `find_unused_parameters=False` as the default (#16611) 2023-02-06 16:51:21 +01:00
test_ddp_strategy.py Add `Strategy.on_exception` (#16646) 2023-02-08 14:00:31 +00:00
test_ddp_strategy_with_comm_hook.py copyright Lightning AI team (#16647) 2023-02-06 15:26:51 +01:00
test_deepspeed_strategy.py Make dataloader_idx optional for batch start/end hooks (#16753) 2023-02-14 18:31:09 +01:00
test_fsdp.py Make dataloader_idx optional for batch start/end hooks (#16753) 2023-02-14 18:31:09 +01:00
test_registry.py Set `find_unused_parameters=False` as the default (#16611) 2023-02-06 16:51:21 +01:00
test_single_device_strategy.py copyright Lightning AI team (#16647) 2023-02-06 15:26:51 +01:00
test_tpu_spawn.py Run XLA's dataloader validation per dataloader (#16775) 2023-02-16 15:09:23 +01:00