search

huggingface / nn_pruning

Prune a model while finetuning or training.
Apache License 2.0
394 stars 58 forks source link

Adding BART architecture #15

Closed echarlaix closed 3 years ago

madlag commented 3 years ago

Thanks, that's a much cleaner way to extend the pruning mechanism.

vblagoje commented 3 years ago

@madlag @echarlaix adding other architectures shouldn't be a lot more complicated than BART, am I overlooking something?