lightning/tests/tests_pytorch/strategies
Carlos Mocholí ec4f592ecf
Sequential `CombinedLoader` to flatten the eval and predict loops (#16726)
2023-02-17 17:37:11 +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 Rename the TPUSpawnStrategy to XLAStrategy (#16781) 2023-02-17 02:06:24 +00:00
test_single_device_strategy.py Sequential `CombinedLoader` to flatten the eval and predict loops (#16726) 2023-02-17 17:37:11 +01:00
test_xla.py Sequential `CombinedLoader` to flatten the eval and predict loops (#16726) 2023-02-17 17:37:11 +01:00