lightning/pytorch_lightning
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
..
accelerators Refactor: legacy accelerators and plugins (#5645) 2021-01-26 20:04:36 -05:00
callbacks [Feat] Adding PruningCallback (#5618) 2021-01-27 01:00:42 -05:00
cluster_environments Refactor: legacy accelerators and plugins (#5645) 2021-01-26 20:04:36 -05:00
core move to Pages dir (#4869) 2021-01-26 15:07:07 -05:00
distributed fix/enable - check F401 (#5201) 2020-12-21 10:15:04 +01:00
loggers docs cleaning - testcode (#5595) 2021-01-26 04:44:54 -05:00
metrics move to Pages dir (#4869) 2021-01-26 15:07:07 -05:00
overrides Refactor LightningDistributedDataParallel (#5185) 2021-01-13 14:35:42 -05:00
plugins Refactor: legacy accelerators and plugins (#5645) 2021-01-26 20:04:36 -05:00
profiler [feat] Add PyTorch Profiler. (#5560) 2021-01-26 06:48:54 -05:00
trainer Refactor: legacy accelerators and plugins (#5645) 2021-01-26 20:04:36 -05:00
tuner Fix isort a few failures (#5504) 2021-01-15 17:44:27 -05:00
utilities [Feat] Adding PruningCallback (#5618) 2021-01-27 01:00:42 -05:00
__init__.py update nightly & upgrade Twine (#5458) 2021-01-26 14:29:47 +01:00
py.typed
setup_tools.py generalise setup tools (#5617) 2021-01-25 23:08:13 -05:00