lightning/tests/plugins
Yi Wang 366fb39d2e
Support post-localSGD in Lightning DDP plugin (#8967)
Co-authored-by: ananthsub <ananth.subramaniam@gmail.com>
Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
2021-08-26 08:24:49 +01:00
..
environments Torch Elastic DDP DeadLock bug fix (#8655) 2021-08-02 21:48:43 +02:00
__init__.py enable ddp as a plugin (#4285) 2020-10-22 05:15:51 -04:00
test_amp_plugins.py Add support for CPU AMP autocast (#9084) 2021-08-25 12:18:00 +00:00
test_checkpoint_io_plugin.py Simplify checkpoint connector loading after Checkpoint IO plugin's introduction (#9045) 2021-08-23 13:12:18 -07:00
test_cluster_integration.py Replace `yapf` with `black` (#7783) 2021-07-26 13:37:35 +02:00
test_custom_plugin.py Add property to skip restoring optimizers and schedulers via plugin (#8644) 2021-07-31 10:08:10 +02:00
test_ddp_fully_sharded_with_full_state_dict.py Replace `yapf` with `black` (#7783) 2021-07-26 13:37:35 +02:00
test_ddp_plugin.py update an outdated error message in DDPPlugin (#9005) 2021-08-23 15:29:07 +00:00
test_ddp_plugin_with_comm_hook.py Support post-localSGD in Lightning DDP plugin (#8967) 2021-08-26 08:24:49 +01:00
test_ddp_spawn_plugin.py Replace `yapf` with `black` (#7783) 2021-07-26 13:37:35 +02:00
test_deepspeed_plugin.py Deprecate `prepare_data_per_node` flag on Trainer and set it as a property for DataHooks (#8958) 2021-08-23 12:43:45 +00:00
test_double_plugin.py Replace `yapf` with `black` (#7783) 2021-07-26 13:37:35 +02:00
test_plugins_registry.py Add DeepSpeed Stage 1 + doc improvements for model parallel (#8974) 2021-08-18 19:40:19 +05:30
test_sharded_plugin.py Replace `yapf` with `black` (#7783) 2021-07-26 13:37:35 +02:00
test_single_device_plugin.py Replace `yapf` with `black` (#7783) 2021-07-26 13:37:35 +02:00
test_tpu_spawn.py Replace `yapf` with `black` (#7783) 2021-07-26 13:37:35 +02:00