lightning/pytorch_lightning
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
..
accelerators Flash predict step (#6577) 2021-03-23 11:13:13 -04:00
callbacks refactoring setup (#6590) 2021-03-22 08:39:19 -04:00
core Flash predict step (#6577) 2021-03-23 11:13:13 -04:00
distributed [bug] Update broadcast + reduce decision ModelCheckpoint] (#6410) 2021-03-14 17:14:27 +00:00
loggers Improve DummyLogger (#6398) 2021-03-09 23:18:38 +00:00
metrics prune metrics: info retrieval (#6649) 2021-03-23 15:05:32 +00:00
overrides Flash predict step (#6577) 2021-03-23 11:13:13 -04:00
plugins Flash predict step (#6577) 2021-03-23 11:13:13 -04:00
profiler Refactor base profilers 3/5 (#6621) 2021-03-23 10:07:35 +00:00
trainer Flash predict step (#6577) 2021-03-23 11:13:13 -04:00
tuner Fix tuner.scale_batch_size not finding batch size attribute when using datamodule (#5968) 2021-03-14 09:16:19 +01:00
utilities fix comparing versions (#6434) 2021-03-23 07:51:45 +00:00
__init__.py refactoring setup (#6590) 2021-03-22 08:39:19 -04:00
info.py refactoring setup (#6590) 2021-03-22 08:39:19 -04:00
py.typed make PyTorch Lightning PEP 561 Compliant (#3187) 2020-09-09 13:37:03 -04:00
setup_tools.py refactoring setup (#6590) 2021-03-22 08:39:19 -04:00