thomas chaton 0995d30fab Flash predict step (#6577) ... * add predict_step * Update predict_loop.py * Update trainer.py * Update trainer.py * resolve bugs * update * update * update * resolve bug * resolve some failing tests * udpate tests * update * resolve tests * add a test * remove typo * add a test for attachement * update * changed to on_train_dataloader * remove __flash_special_attr__ * resolve tests * update * update * update * update on comments * Update pytorch_lightning/trainer/data_loading.py Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com> Co-authored-by: Justus Schock <12886177+justusschock@users.noreply.github.com> Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>		2021-03-23 11:13:13 -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	Hardware specific parts of Accelerator Refactoring (#5719)	2021-02-01 08:34:59 -05: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