Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
lightning/pytorch_lightning/utilities
History
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 [Feat] Adding PruningCallback (#5618) 2021-01-27 01:00:42 -05:00
apply_func.py formatting to PL utils (#5713) 2021-01-30 15:28:59 +01:00
argparse.py formatting to PL utils (#5713) 2021-01-30 15:28:59 +01:00
argparse_utils.py formatting to PL utils (#5713) 2021-01-30 15:28:59 +01:00
cloud_io.py Update cloud_io.py (#4936) 2020-12-04 11:00:06 +01:00
data.py formatting to PL utils (#5713) 2021-01-30 15:28:59 +01:00
debugging.py formatting to PL utils (#5713) 2021-01-30 15:28:59 +01:00
device_dtype_mixin.py fix docs render (#5610) 2021-01-25 20:21:00 -05:00
device_parser.py formatting to PL utils (#5713) 2021-01-30 15:28:59 +01:00
distributed.py formatting to PL utils (#5713) 2021-01-30 15:28:59 +01:00
enums.py unify LightningEnum (#5389) 2021-01-12 08:56:07 -05:00
exceptions.py formatting to PL utils (#5713) 2021-01-30 15:28:59 +01:00
imports.py [Feat] Adding PruningCallback (#5618) 2021-01-27 01:00:42 -05:00
memory.py [Fix] Move log value to cpu. (#4592) 2020-11-10 21:13:41 +00:00
model_helpers.py Fix isort failures in utilities (#5530) 2021-01-15 13:57:40 -05:00
model_utils.py formatting to PL utils (#5713) 2021-01-30 15:28:59 +01:00
parsing.py formatting to PL utils (#5713) 2021-01-30 15:28:59 +01:00
seed.py formatting to PL utils (#5713) 2021-01-30 15:28:59 +01:00
upgrade_checkpoint.py notices (#4118) 2020-10-13 07:18:07 -04:00
warning_utils.py formatting to PL utils (#5713) 2021-01-30 15:28:59 +01:00
warnings.py Refactor LightningDataParallel (#5670) 2021-01-31 06:08:16 -05:00
xla_device.py formatting to PL utils (#5713) 2021-01-30 15:28:59 +01:00
xla_device_utils.py formatting to PL utils (#5713) 2021-01-30 15:28:59 +01:00