huggingface / nn_pruning

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

Fix quantization tests #38

Open LukasHedegaard opened 2 years ago

LukasHedegaard commented 2 years ago

This PR fixes the nn_pruning/tests/test_quantization.py tests. How: