thomas chaton e147127c0e [feat] Add better support for predict + ddp 2/3 (#7215) ... * wip * update * update * update * update * update * typo * update on comments * update * update * update * update * update changelog * update * Fix merge * Fix merge * move code * resolve test * add extra test * add an extra test * update on comments * add typing * resolve flake8 * Refactor and Docs * Fix tests * Fix tests * Fix tests * Duplicate * Fix tests * resolve bug * update * update on comments * update * update changelog * update * update * remove tpu * resolve flake8 * update on comments * update on comments * update on comment * resolve flake8 * add a cpu test for predict * add None test * update * Update CHANGELOG.md Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com> * resolve tests Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>		2021-04-27 08:46:45 -04: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	[feat] Add better support for predict + ddp 2/3 (#7215)	2021-04-27 08:46:45 -04: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