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	CI: add flake8 (#4239)	2020-10-19 21:20:17 +01:00
datamodule.py	argparse: Add use_argument_group=True (#6088)	2021-03-11 10:50:49 -05:00
decorators.py	Add before_batch_transfer and after_batch_transfer hooks (#3671)	2021-02-18 06:58:12 -05:00
grads.py	formatting 5/n: Core (#5721)	2021-02-08 14:29:43 -05:00
hooks.py	Flash predict step (#6577)	2021-03-23 11:13:13 -04:00
lightning.py	Flash predict step (#6577)	2021-03-23 11:13:13 -04:00
memory.py	Handle torch.jit scripted modules in layer summary (#6511)	2021-03-15 03:17:42 +01:00
optimizer.py	[bugfix] Check LightningOptimizer doesn't delete optimizer hooks (#6305)	2021-03-04 20:11:59 +00:00
saving.py	fix duplicate console logging bug v2 (#6275)	2021-03-02 15:17:55 +05:30
step_result.py	prune warning & deprecation wrapper (#6540)	2021-03-16 14:55:31 +00:00