lightning/tests/plugins
Danielle Pintz b3a5c7f442
Add `enable_progress_bar` to Trainer constructor (#9664)
2021-09-24 22:53:31 -07:00
..
environments Skip reconciliate_processes if used within a cluster environment that creates processes externally (#9389) 2021-09-15 11:54:17 +01:00
__init__.py
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 Remove `call_configure_sharded_model` lifecycle property (#9612) 2021-09-24 03:57:53 +02: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 Move add_to_queue/get_from_queue to DDPSpawnPlugin (#9118) 2021-09-10 20:58:02 +00:00
test_deepspeed_plugin.py Add `enable_progress_bar` to Trainer constructor (#9664) 2021-09-24 22:53:31 -07: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 Fix gradient accumulation for `ShardedDataParallel` (#9122) 2021-09-22 08:56:38 +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