lightning/pytorch_lightning/overrides
Adrian Wälchli 80c5293514
fix self.device access in DataParallel (#6414)
2021-04-13 02:03:24 +02:00
..
__init__.py Refactor LightningDataParallel (#5670) 2021-01-31 06:08:16 -05:00
base.py Flash predict step (#6577) 2021-03-23 11:13:13 -04:00
data_parallel.py fix self.device access in DataParallel (#6414) 2021-04-13 02:03:24 +02:00
distributed.py Refactor LightningDataParallel (#5670) 2021-01-31 06:08:16 -05:00
fairscale.py PoC: Accelerator refactor (#5743) 2021-02-12 15:48:56 -05:00
torch_distributed.py [bug] Update broadcast + reduce decision ModelCheckpoint] (#6410) 2021-03-14 17:14:27 +00:00