lightning/pytorch_lightning/trainer
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
..
connectors Flash predict step (#6577) 2021-03-23 11:13:13 -04:00
__init__.py added trainer api docs (#4569) 2020-11-07 14:18:45 -05:00
callback_hook.py Add outputs param for `on_val/test_epoch_end` hooks (#6120) 2021-03-16 12:15:16 -04:00
configuration_validator.py Add trainer.predict config validation (#6543) 2021-03-21 21:07:54 +00:00
data_loading.py Flash predict step (#6577) 2021-03-23 11:13:13 -04:00
deprecated_api.py Remove unused mixin attributes (#6487) 2021-03-12 08:29:52 +00:00
evaluation_loop.py Refactor base profilers 3/5 (#6621) 2021-03-23 10:07:35 +00:00
ignored_warnings.py formatting 4/n: Trainer (#5720) 2021-02-03 09:25:42 +00:00
logging.py Remove unused mixin attributes (#6487) 2021-03-12 08:29:52 +00:00
model_hooks.py Pass {fit,validate,test,predict} to setup() and teardown() (#6386) 2021-03-08 15:27:07 +01:00
optimizers.py Fix trainer not resetting lightning_optimizers (#6372) 2021-03-08 09:58:03 +08:00
predict_loop.py Flash predict step (#6577) 2021-03-23 11:13:13 -04:00
properties.py [refactor] Add setup to profilers + _run_stage_setup to trainer 2/5 (#6633) 2021-03-22 14:32:31 -04:00
states.py Add Trainer.validate(…) method to run one validation epoch (#4948) 2021-03-11 03:46:37 +01:00
supporters.py formatting 4/n: Trainer (#5720) 2021-02-03 09:25:42 +00:00
trainer.py Flash predict step (#6577) 2021-03-23 11:13:13 -04:00
training_loop.py Refactor base profilers 3/5 (#6621) 2021-03-23 10:07:35 +00:00
training_tricks.py Remove unused mixin attributes (#6487) 2021-03-12 08:29:52 +00:00