Carlos Mocholí
|
38ad9e0764
|
[ModelPruning] Add missing attribute with use_global_unstructured=False and verbose (#6045)
|
2021-02-18 10:40:34 +00:00 |
|
Akihiro Nitta
|
0a2fb05aac
|
Document exceptions in callbacks (#5541)
* Add Raises: section to docstring
* Add Raises section to the docs
* Add raises section to the docs
* Apply suggestions from code review
Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
* fix
* Remove unnecessary instance check
Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
|
2021-02-15 10:24:36 +00:00 |
|
Carlos Mocholí
|
a028171f26
|
Fix Pruning callback and add a few features (#5825)
* Remove pruning check because it was added in 1.4.0 and that is our minimal torch version
* Fixing many bugs
* Fix misconfig test
* Fix tests
* Improve error message
* Reduce whitespace
* WIP
* TODOs
* _MODULE_CONTAINERS
* Add LTH test
* Allow resampling
* Iterative pruning
* Log pruning percentage
* Properly make pruning permanent
* Fix docstring
* Minor changes
* Test loading non-permanent model
* corrent bugs
* Revert "corrent bugs"
This reverts commit ffb8d47547.
* Add beta warning
* Fix docs
* 2 verbosity levels
* OCD
Co-authored-by: Your Name <you@example.com>
|
2021-02-10 15:03:23 +00:00 |
|
Jirka Borovec
|
79d42d83e7
|
formatting 3/n: PL modules (#5716)
* cb
* log
* prof
* tune
* flake8
|
2021-02-08 14:28:38 -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 |