lightning/tests/tests_pytorch/strategies
Carlos Mocholí bf51844917
Remove memory-retaining epoch-end hooks (#16520)
2023-02-06 17:00:36 +00: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_bagua_strategy.py copyright Lightning AI team (#16647) 2023-02-06 15:26:51 +01:00
test_colossalai.py copyright Lightning AI team (#16647) 2023-02-06 15:26:51 +01:00
test_common.py copyright Lightning AI team (#16647) 2023-02-06 15:26:51 +01: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 copyright Lightning AI team (#16647) 2023-02-06 15:26:51 +01: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 copyright Lightning AI team (#16647) 2023-02-06 15:26:51 +01: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 Remove memory-retaining epoch-end hooks (#16520) 2023-02-06 17:00:36 +00:00
test_dp.py Remove memory-retaining epoch-end hooks (#16520) 2023-02-06 17:00:36 +00:00
test_fsdp.py tests: switch imports for pytorch (#16595) 2023-02-02 10:06:45 +00: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 copyright Lightning AI team (#16647) 2023-02-06 15:26:51 +01:00