lightning/pytorch_lightning/trainer/connectors
chaton 0435e23a64 deprecate enable_pl_optimizer as it is not restored properly (#5244)
* update

* clean test

* still in progress

* udpdate test

* update

* update

* resolve flake

* add test for zero_grad

* update

* works without accumulated_grad

* update

* update

* resolve amp

* revert back to True

* update

* clean tests

* cleaned out

* typo

* update test

* git repare bug

* remove print

* udpate

* Fix formatting/optimizer imports

* Refactor the test for cleanliness

* Add vanilla model to the test, better var names

* Fixed var names, let's clean up these mock tests

* repare test

* update test

* resolve flake8

* add manual_optimization

* update tests

* resolve flake8

* add random accumulate_grad_batches

* improve test

* Update tests/trainer/optimization/test_parity_automatic_optimization.py

Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>

* Update tests/trainer/optimization/test_parity_automatic_optimization.py

Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>

* update

* clean tests

* correct bug

* Apply suggestions from code review

* format

* adress comments

* update on comments

* wip

* typo

* depreceate enable_pl_optimizer

* resolve latest bugs

* update

* resolve merge

* add comment

* Update pytorch_lightning/core/lightning.py

Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>

* Update tests/deprecated_api/test_remove_1-3.py

Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>

* Update pytorch_lightning/trainer/connectors/optimizer_connector.py

Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>

* Update pytorch_lightning/trainer/trainer.py

Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>

* Update pytorch_lightning/trainer/trainer.py

Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>

* Update tests/trainer/optimization/test_parity_automatic_optimization.py

Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>

* update on comments

* update restore

* add a property

* remove setstate as not needed anymore

* update test

* provide optimizer to on_before_zero_grad

* update on comments

* update on comments

* Update pytorch_lightning/trainer/trainer.py

Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>

* Update tests/trainer/optimization/test_parity_automatic_optimization.py

Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>

* Update tests/trainer/optimization/test_parity_automatic_optimization.py

Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>

* Update tests/trainer/optimization/test_parity_automatic_optimization.py

Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>

* mofidy import

* update changelog

* resolve flake8

* update

* update

* clean doc

Co-authored-by: SeanNaren <sean@grid.ai>
Co-authored-by: Ubuntu <ubuntu@ip-172-31-62-109.ec2.internal>
Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
Co-authored-by: Jirka Borovec <jirka.borovec@seznam.cz>
Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
Co-authored-by: Sean Naren <sean.narenthiran@gmail.com>

(cherry picked from commit f2e99d617f)
2021-01-26 14:29:46 +01:00
..
logger_connector [bugfix] Logging only on `not should_accumulate()` during training (#5417) 2021-01-26 14:28:47 +01:00
__init__.py ref: group connectors (#3472) 2020-09-11 23:33:09 -04:00
callback_connector.py Set progressbar refresh rate in Google Colab (#5516) 2021-01-19 12:47:14 -05:00
checkpoint_connector.py Fix isort failures in trainer (#5529) 2021-01-18 13:42:50 -05:00
data_connector.py Fix isort failures in trainer (#5529) 2021-01-18 13:42:50 -05:00
debugging_connector.py Disable checkpointing, earlystopping and logging with fast_dev_run (#5277) 2021-01-06 12:57:24 +01:00
env_vars_connector.py Fix isort failures in trainer (#5529) 2021-01-18 13:42:50 -05:00
model_connector.py Refactor: clean trainer device & distrib getters (#5300) 2021-01-12 05:22:37 -05:00
optimizer_connector.py deprecate enable_pl_optimizer as it is not restored properly (#5244) 2021-01-26 14:29:46 +01:00
precision_connector.py deprecate enable_pl_optimizer as it is not restored properly (#5244) 2021-01-26 14:29:46 +01:00
profiler_connector.py [feat] Add PyTorch Profiler. (#5560) 2021-01-26 06:48:54 -05:00
slurm_connector.py Fix isort failures in trainer (#5529) 2021-01-18 13:42:50 -05:00
training_trick_connector.py Fix isort failures in trainer (#5529) 2021-01-18 13:42:50 -05:00