lightning/pytorch_lightning/plugins/legacy
Adrian Wälchli 692f77b8a7
Refactor LightningDataParallel (#5670)
* module

* fix model access

* scalar conversion

* refactor

* kwargs

* auto unsqueeze

* refactor code duplication

* clean up

* docs

* update dp docs

* changelog

* generalize test

* test

* rename

* warning cache

* isort

* unsqueezing test

* device

* device

* scalar test

* device

* device

* include coverage of overrides

* clear

* add deprecation test

* docs

* improve coverage

* increase coverage

* fix merge

* extend test

* rename base class

* mention the predict method in docs

* combine iteration over collection

* remove override

* move

* line

* Apply suggestions from code review

* fix running stage

* f401

* fix cyclic import

Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
2021-01-31 06:08:16 -05:00
..
__init__.py Refactor: legacy accelerators and plugins (#5645) 2021-01-26 20:04:36 -05:00
apex.py Refactor: legacy accelerators and plugins (#5645) 2021-01-26 20:04:36 -05:00
ddp_plugin.py Refactor LightningDataParallel (#5670) 2021-01-31 06:08:16 -05:00
ddp_sequential_plugin.py Fix isort failures in core (#5526) 2021-01-27 05:02:16 -05:00
native_amp.py Refactor: legacy accelerators and plugins (#5645) 2021-01-26 20:04:36 -05:00
plugin.py Refactor: legacy accelerators and plugins (#5645) 2021-01-26 20:04:36 -05:00
plugin_connector.py Refactor: legacy accelerators and plugins (#5645) 2021-01-26 20:04:36 -05:00
precision_plugin.py Refactor: legacy accelerators and plugins (#5645) 2021-01-26 20:04:36 -05:00
rpc_plugin.py Refactor: legacy accelerators and plugins (#5645) 2021-01-26 20:04:36 -05:00
sharded_native_amp_plugin.py Fix isort failures in core (#5526) 2021-01-27 05:02:16 -05:00
sharded_plugin.py Refactor: legacy accelerators and plugins (#5645) 2021-01-26 20:04:36 -05:00