lightning/pytorch_lightning/strategies
Rohit Gupta 06b8f82b8a
Update API references in doc (#11357)
2022-01-07 15:56:17 +01:00
..
__init__.py Introduce StrategyRegistry (#11233) 2022-01-05 17:14:18 +05:30
ddp.py Introduce StrategyRegistry (#11233) 2022-01-05 17:14:18 +05:30
ddp2.py Introduce strategies directory for Training Strategies (#11226) 2021-12-22 20:23:30 +00:00
ddp_spawn.py Introduce StrategyRegistry (#11233) 2022-01-05 17:14:18 +05:30
deepspeed.py Add Accelerators section to Lightning docs (#10755) 2022-01-06 19:12:44 +05:30
dp.py Introduce strategies directory for Training Strategies (#11226) 2021-12-22 20:23:30 +00:00
fully_sharded.py Fix exception message for FSDP running on CPU (#11325) 2022-01-05 18:02:31 +01:00
horovod.py Introduce strategies directory for Training Strategies (#11226) 2021-12-22 20:23:30 +00:00
ipu.py Update API references in doc (#11357) 2022-01-07 15:56:17 +01:00
parallel.py Rename training_type_plugin file to strategy (#11239) 2021-12-23 14:01:23 +00:00
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 Rename training_type_plugin file to strategy (#11239) 2021-12-23 14:01:23 +00:00
single_tpu.py Introduce strategies directory for Training Strategies (#11226) 2021-12-22 20:23:30 +00:00
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 Introduce strategies directory for Training Strategies (#11226) 2021-12-22 20:23:30 +00:00