Jirka Borovec
|
a0f7831278
|
fix miss-leading imports in tests (#5873)
* fix imorts
* .
|
2021-02-09 05:10:52 -05:00 |
|
Jirka Borovec
|
91f63deabc
|
formatting tests: 5/5 (#5848)
* cb
* acc
* plug
* .
|
2021-02-06 07:28:26 -05:00 |
|
Jirka Borovec
|
f83cca6107
|
formatting flake8 & isort (#5824)
* formatting
* isort
* make
* yapf
* isort
|
2021-02-05 18:33:12 -05:00 |
|
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 |