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/callbacks
History
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
..
__init__.py [BugOnFeat] Resolve bug with Finetuning (#5744) 2021-02-04 18:36:54 +00:00
base.py formatting 3/n: PL modules (#5716) 2021-02-08 14:28:38 -05:00
early_stopping.py formatting 3/n: PL modules (#5716) 2021-02-08 14:28:38 -05:00
finetuning.py formatting 3/n: PL modules (#5716) 2021-02-08 14:28:38 -05:00
gpu_stats_monitor.py formatting 3/n: PL modules (#5716) 2021-02-08 14:28:38 -05:00
gradient_accumulation_scheduler.py formatting 3/n: PL modules (#5716) 2021-02-08 14:28:38 -05:00
lambda_function.py formatting 3/n: PL modules (#5716) 2021-02-08 14:28:38 -05:00
lr_monitor.py formatting 3/n: PL modules (#5716) 2021-02-08 14:28:38 -05:00
model_checkpoint.py formatting 3/n: PL modules (#5716) 2021-02-08 14:28:38 -05:00
progress.py formatting 3/n: PL modules (#5716) 2021-02-08 14:28:38 -05:00
pruning.py Fix Pruning callback and add a few features (#5825) 2021-02-10 15:03:23 +00:00