lightning/pytorch_lightning/overrides
Adrian Wälchli e806bb77fa
Refactor LightningDistributedDataParallel (#5185)
* add wrapper

* add squeeze

* replace LightningDistributedDP

* update import

* module access

* inputs

* refactor warning

* update

* resolve flake8

* remove old class

* set find unused params to False

* update docstrings

* update docs

* update docs

* add changelog

* deprecation

* rename wrapper -> module

* rename pl_module

* add unit tests

* Revert "add changelog"

This reverts commit 02ec0a6864f4ba2ace3bb6fc6ebc364e1a80ffd7.

* Revert "set find unused params to False"

This reverts commit 8e451515e6ba3227d00f4a5cb63f332cfedb7b30.

Co-authored-by: Ubuntu <thomas@grid.ai>
2021-01-13 14:35:42 -05:00
..
__init__.py Refactor: name modules (#548) 2019-11-26 22:39:18 -05:00
data_parallel.py Refactor LightningDistributedDataParallel (#5185) 2021-01-13 14:35:42 -05:00
fairscale.py set xxx_AVAILABLE as protected (#5082) 2020-12-14 20:19:05 +05:30