lightning/tests/plugins
Sean Naren c66cd12445
Remove partitioning of model in ZeRO 3 (#10655)
2021-12-17 12:36:53 +00:00
..
environments Fix typing in `pl.plugins.environments` (#10943) 2021-12-07 02:14:02 +00:00
__init__.py
test_amp_plugins.py Rename special to standalone (#10779) 2021-11-26 17:13:14 +00:00
test_checkpoint_io_plugin.py Update strategy flag in docs (#10000) 2021-10-20 21:02:53 +05:30
test_cluster_integration.py Update strategy flag in docs (#10000) 2021-10-20 21:02:53 +05:30
test_custom_plugin.py Unify checkpoint load paths [redo #9693] (#10061) 2021-10-25 19:05:31 +00:00
test_ddp_fully_sharded_with_full_state_dict.py 3/n Move accelerator into Strategy (#11022) 2021-12-16 04:41:34 +00:00
test_ddp_plugin.py 3/n Move accelerator into Strategy (#11022) 2021-12-16 04:41:34 +00:00
test_ddp_plugin_with_comm_hook.py 3/n Move accelerator into Strategy (#11022) 2021-12-16 04:41:34 +00:00
test_ddp_spawn_plugin.py 2/n Simplify spawn plugins: Spawn immediately (#10896) 2021-12-09 18:56:24 +00:00
test_deepspeed_plugin.py Remove partitioning of model in ZeRO 3 (#10655) 2021-12-17 12:36:53 +00:00
test_double_plugin.py Drop torch 1.6 testing (#10390) 2021-11-13 20:35:03 +00:00
test_plugins_registry.py Add more trainer config tests (#10319) 2021-11-05 10:42:58 +01:00
test_sharded_plugin.py 3/n Move accelerator into Strategy (#11022) 2021-12-16 04:41:34 +00:00
test_single_device_plugin.py Replace `yapf` with `black` (#7783) 2021-07-26 13:37:35 +02:00
test_tpu_spawn.py Remove leftover `clean_logger` call in tests (#11080) 2021-12-17 00:23:32 +00:00