lightning/pytorch_lightning/accelerators/legacy
chaton 3da28fd634
[feat] 1/2 Add trainer.predict (#5579)
* start adding predict

* add predict

* resolve test

* add predict

* remove limit_predict

* update

* add test for predict

* typo

* update on comments

* remove predict_step

* update ddp_shareded

* check ddp_sharded

* resolve on comments

* resolve isort

* update dp

* add test dp 1 gpu

* made default forward

* resolve path

* resolve bug

* update on comments

* resolve doc

* resolve bug

* update

* resolve bug

* update on comments

* resolve pep8

* update test doc

* update on comments

* solve special tests

* resolve bug

* resolve flake8

* Update pytorch_lightning/callbacks/progress.py

Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>

* Update pytorch_lightning/trainer/trainer.py

Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>

* add predict to LightningModule

* missing predict

* typo

* rename is_prediction to _predicting

* add

* update

* update

* update doc

Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
Co-authored-by: mergify[bot] <37929162+mergify[bot]@users.noreply.github.com>
2021-01-27 11:38:14 -05:00
..
__init__.py Refactor: legacy accelerators and plugins (#5645) 2021-01-26 20:04:36 -05:00
accelerator.py Refactor: legacy accelerators and plugins (#5645) 2021-01-26 20:04:36 -05:00
accelerator_connector.py Refactor: legacy accelerators and plugins (#5645) 2021-01-26 20:04:36 -05:00
cpu_accelerator.py [feat] 1/2 Add trainer.predict (#5579) 2021-01-27 11:38:14 -05:00
ddp2_accelerator.py [feat] 1/2 Add trainer.predict (#5579) 2021-01-27 11:38:14 -05:00
ddp_accelerator.py [feat] 1/2 Add trainer.predict (#5579) 2021-01-27 11:38:14 -05:00
ddp_cpu_hpc_accelerator.py Refactor: legacy accelerators and plugins (#5645) 2021-01-26 20:04:36 -05:00
ddp_cpu_spawn_accelerator.py [feat] 1/2 Add trainer.predict (#5579) 2021-01-27 11:38:14 -05:00
ddp_hpc_accelerator.py [feat] 1/2 Add trainer.predict (#5579) 2021-01-27 11:38:14 -05:00
ddp_spawn_accelerator.py [feat] 1/2 Add trainer.predict (#5579) 2021-01-27 11:38:14 -05:00
dp_accelerator.py [feat] 1/2 Add trainer.predict (#5579) 2021-01-27 11:38:14 -05:00
gpu_accelerator.py [feat] 1/2 Add trainer.predict (#5579) 2021-01-27 11:38:14 -05:00
horovod_accelerator.py [feat] 1/2 Add trainer.predict (#5579) 2021-01-27 11:38:14 -05:00
tpu_accelerator.py [feat] 1/2 Add trainer.predict (#5579) 2021-01-27 11:38:14 -05:00