lightning/tests/strategies
four4fish 15364c18c8
Check `parallel_devices` passed through `strategy` is consistent with the `accelerator` flag (#12105)
2022-03-03 10:30:24 -08:00
..
__init__.py
ddp_model.py Move strategy tests from accelerators to strategies directory (#11329) 2022-02-22 05:14:18 +00:00
test_bagua_strategy.py Add `Trainer(strategy="bagua")` (#11146) 2022-02-04 17:02:09 +00:00
test_common.py Move strategy tests from accelerators to strategies directory (#11329) 2022-02-22 05:14:18 +00:00
test_custom_plugin.py Add `SyncBatchNormPlugin` (#11754) 2022-03-01 19:41:40 +05:30
test_ddp.py Fix passing _ddp_params_and_buffers_to_ignore (#11949) 2022-02-24 17:22:48 +00:00
test_ddp_fully_sharded_with_full_state_dict.py Fix exception message for FSDP running on CPU (#11325) 2022-01-05 18:02:31 +01:00
test_ddp_spawn.py Move strategy tests from accelerators to strategies directory (#11329) 2022-02-22 05:14:18 +00:00
test_ddp_spawn_strategy.py Add process launchers (#11643) 2022-02-17 21:16:51 +00:00
test_ddp_strategy.py Check `parallel_devices` passed through `strategy` is consistent with the `accelerator` flag (#12105) 2022-03-03 10:30:24 -08:00
test_ddp_strategy_with_comm_hook.py
test_deepspeed_strategy.py Rewrite accelerator_connector (#11448) 2022-02-17 23:38:39 +00:00
test_dp.py Move strategy tests from accelerators to strategies directory (#11329) 2022-02-22 05:14:18 +00:00
test_sharded_strategy.py
test_single_device_strategy.py Remove `Strategy.on_tpu` property (#11536) 2022-01-20 08:25:26 +01:00
test_strategy_registry.py Rewrite accelerator_connector (#11448) 2022-02-17 23:38:39 +00:00
test_tpu_spawn.py fix typos (#11937) 2022-02-16 17:27:51 -08:00