lightning/tests/plugins
Carlos Mocholí 4a64bc3fd3
Fix DeepSpeed lr scheduler logic (#8527)
* Fix deepspeed scheduler logic

* Fix tests

* Minor changes

* Improve tests

* inference fix

* CHANGELOG

Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
2021-07-23 10:08:58 +01:00
..
environments support launching Lightning ddp with traditional command (#7480) 2021-07-14 11:25:36 +00:00
__init__.py
test_amp_plugins.py Add the `on_before_optimizer_step` hook (#8048) 2021-07-09 13:30:52 +02:00
test_cluster_integration.py Use `is` to compare type of objects (#8404) 2021-07-19 11:17:45 +00:00
test_custom_plugin.py
test_ddp_fully_sharded_with_full_state_dict.py FSDP with full state dict (#7487) 2021-05-24 08:11:45 +01:00
test_ddp_plugin.py fix NCCL error with non-consecutive trainer gpus (#8165) 2021-06-28 22:08:10 +02:00
test_ddp_plugin_with_comm_hook.py
test_ddp_spawn_plugin.py Add `add_to_queue`/`get_from_queue` for DDP spawn(#7916) 2021-06-23 03:19:37 +02:00
test_deepspeed_plugin.py Fix DeepSpeed lr scheduler logic (#8527) 2021-07-23 10:08:58 +01:00
test_double_plugin.py Fix double precision casting complex buffers (#8208) 2021-06-30 10:57:42 +01:00
test_plugins_registry.py Add `tpu_spawn_debug` to plugin registry (#7933) 2021-06-15 22:32:51 +00:00
test_sharded_plugin.py Fix Special Tests (#7841) 2021-06-16 19:39:03 +02:00
test_single_device_plugin.py Merge pull request #7872 from PyTorchLightning/refactor/logger-poc-changes 2021-06-08 09:04:16 -04:00
test_tpu_spawn.py Standardize positional datamodule and argument names (#7431) 2021-06-15 11:50:13 +00:00