dynamic-pruning
Codes for Dynamic Network Pruning with Interpretable Layerwise Channel Selection
Usage
- train_baseline.py: pretrain baseline models
- main.py: train dynamic models with decision units
- finetune.py: pruning models based on dynamic pruning method