lightning/pytorch_lightning/utilities
chaton d0aaf983b9
[Feat] Adding PruningCallback (#5618)
* wip

* add pruning callback

* add condition for duplicated weights

* update on comments

* update on comments

* update on comments

* add more tests

* resolve flake8

* resolve on comments

* update changelog

* update on comments

* update on comments

* change order

* remove ddp_spawn skip

* update

* typo

* Update pytorch_lightning/callbacks/pruning.py

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

* Update pytorch_lightning/callbacks/pruning.py

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

* update on comments

* forgot platform

* update on comments

* remove     @rank_zero_only

Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
Co-authored-by: mergify[bot] <37929162+mergify[bot]@users.noreply.github.com>
2021-01-27 01:00:42 -05:00
..
__init__.py [Feat] Adding PruningCallback (#5618) 2021-01-27 01:00:42 -05:00
apply_func.py [bug-fix] Call transfer_batch_to_device in DDPlugin (#5195) 2021-01-26 14:28:45 +01:00
argparse.py Fix isort failures in utilities (#5530) 2021-01-15 13:57:40 -05:00
argparse_utils.py Unify names in Utils (#5199) 2020-12-22 00:23:33 +01:00
cloud_io.py Update cloud_io.py (#4936) 2020-12-04 11:00:06 +01:00
data.py Fix isort failures in utilities (#5530) 2021-01-15 13:57:40 -05:00
debugging.py Fix isort failures in utilities (#5530) 2021-01-15 13:57:40 -05:00
device_dtype_mixin.py fix docs render (#5610) 2021-01-25 20:21:00 -05:00
device_parser.py Raise TypeError instead of using asserting with condition of types. (#5536) 2021-01-19 07:39:43 -05:00
distributed.py fix docs render (#5610) 2021-01-25 20:21:00 -05:00
enums.py unify LightningEnum (#5389) 2021-01-12 08:56:07 -05:00
exceptions.py added copyright notices (#3062) 2020-08-19 22:03:22 -04: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 Unify names in Utils (#5199) 2020-12-22 00:23:33 +01:00
parsing.py Fix merge _module_available 2021-01-26 14:29:47 +01:00
seed.py fix imports / isort / flake8 2021-01-26 14:57:34 +01:00
upgrade_checkpoint.py notices (#4118) 2020-10-13 07:18:07 -04:00
warning_utils.py Unify names in Utils (#5199) 2020-12-22 00:23:33 +01:00
warnings.py Unify names in Utils (#5199) 2020-12-22 00:23:33 +01:00
xla_device.py mark todo exceptions (#5320) 2021-01-04 09:07:56 +01:00
xla_device_utils.py Unify names in Utils (#5199) 2020-12-22 00:23:33 +01:00