lightning/pytorch_lightning/callbacks
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
base.py Feat: Add BackboneLambdaFinetunningCallback (#5377) 2021-01-08 16:33:05 -05:00
early_stopping.py Throw MisconfigurationError on unknown mode (#5255) 2021-01-12 02:31:26 -05:00
finetuning.py fix docs render (#5610) 2021-01-25 20:21:00 -05:00
gpu_stats_monitor.py Fix isort a few failures (#5504) 2021-01-15 17:44:27 -05:00
gradient_accumulation_scheduler.py Simplify optimization Logic (#4984) 2020-12-07 12:55:49 +00:00
lambda_function.py Add LambdaCallback (#5347) 2021-01-13 04:42:49 -05:00
lr_monitor.py Allow log_momentum for adaptive optimizers (#5333) 2021-01-06 12:58:34 +01:00
model_checkpoint.py Start version suffixes at 1 (#5008) 2021-01-26 17:29:34 -05:00
progress.py move to Pages dir (#4869) 2021-01-26 15:07:07 -05:00
pruning.py [Feat] Adding PruningCallback (#5618) 2021-01-27 01:00:42 -05:00