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
..
environments	introduce default cluster environment for lightning-specific ddp (#5915)	2021-03-05 01:47:29 +00:00
precision	Add AMP for validation, prediction and testing (#6565)	2021-03-20 23:15:49 +00:00
training_type	Flash predict step (#6577)	2021-03-23 11:13:13 -04:00
__init__.py	DeepSpeed Integration (#5954)	2021-02-17 15:23:42 -05:00
base_plugin.py	Add mypy typing to precision plugins. (#6149)	2021-02-26 14:27:16 +01:00