Rohit Gupta 82c8875f33 Add `LightningModule.lr_scheduler_step` (#10249) ... Co-authored-by: Carlos Mocholi <carlossmocholi@gmail.com>		2022-01-12 03:53:49 +00:00
..
__init__.py	Introduce StrategyRegistry (#11233)	2022-01-05 17:14:18 +05:30
ddp.py	Add DETAIL logs for batch use cases (#11008)	2022-01-12 01:22:48 +01:00
ddp2.py	…
ddp_spawn.py	Introduce StrategyRegistry (#11233)	2022-01-05 17:14:18 +05:30
deepspeed.py	Add `LightningModule.lr_scheduler_step` (#10249)	2022-01-12 03:53:49 +00:00
dp.py	…
fully_sharded.py	Add DETAIL logs for batch use cases (#11008)	2022-01-12 01:22:48 +01:00
horovod.py	Add `LightningModule.lr_scheduler_step` (#10249)	2022-01-12 03:53:49 +00:00
ipu.py	Update API references in doc (#11357)	2022-01-07 15:56:17 +01:00
parallel.py	…
sharded.py	Introduce StrategyRegistry (#11233)	2022-01-05 17:14:18 +05:30
sharded_spawn.py	Introduce StrategyRegistry (#11233)	2022-01-05 17:14:18 +05:30
single_device.py	…
single_tpu.py	…
strategy.py	Fix restoring lr scheduler states with deepspeed strategy (#11322)	2022-01-06 12:34:16 +00:00
strategy_registry.py	Introduce StrategyRegistry (#11233)	2022-01-05 17:14:18 +05:30
tpu_spawn.py	Introduce StrategyRegistry (#11233)	2022-01-05 17:14:18 +05:30
utils.py	…