lightning/tests/plugins
Kaushik B d028e36946
Add remove_checkpoint to CheckpointIO plugin to simplify ModelCheckpo… (#9373)
Co-authored-by: ananthsub <ananth.subramaniam@gmail.com>
Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
Co-authored-by: ananthsub <ananth.subramaniam@gmail.com>
Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
2021-09-10 11:55:04 +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 CI: precommit - docformatter (#8584) 2021-09-06 12:49:09 +00:00
test_checkpoint_io_plugin.py Add remove_checkpoint to CheckpointIO plugin to simplify ModelCheckpo… (#9373) 2021-09-10 11:55:04 +01: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 CI: precommit - docformatter (#8584) 2021-09-06 12:49:09 +00:00
test_ddp_plugin.py Avoid wrapping LightningModule in DDP plugins when not fitting (#9096) 2021-09-02 02:23:59 +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 Avoid wrapping LightningModule in DDP plugins when not fitting (#9096) 2021-09-02 02:23:59 +00:00
test_deepspeed_plugin.py Add a warning to deepspeed when inferring batch size (#9221) 2021-09-07 16:24:00 +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 CI: precommit - docformatter (#8584) 2021-09-06 12:49:09 +00: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