lightning/tests/strategies
Adrian Wälchli 9c8f52ccd1
Fix restoring lr scheduler states with deepspeed strategy (#11322)
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
Co-authored-by: thomas chaton <thomas@grid.ai>
2022-01-06 12:34:16 +00:00
..
__init__.py
test_custom_plugin.py Fix restoring lr scheduler states with deepspeed strategy (#11322) 2022-01-06 12:34:16 +00:00
test_ddp_fully_sharded_with_full_state_dict.py Fix exception message for FSDP running on CPU (#11325) 2022-01-05 18:02:31 +01:00
test_ddp_spawn_strategy.py
test_ddp_strategy.py
test_ddp_strategy_with_comm_hook.py
test_deepspeed_strategy.py Fix restoring lr scheduler states with deepspeed strategy (#11322) 2022-01-06 12:34:16 +00:00
test_sharded_strategy.py
test_single_device_strategy.py
test_strategy_registry.py Introduce StrategyRegistry (#11233) 2022-01-05 17:14:18 +05:30
test_tpu_spawn.py