he-y / soft-filter-pruning

Soft Filter Pruning for Accelerating Deep Convolutional Neural Networks
https://arxiv.org/abs/1808.06866
374 stars 74 forks source link

Pruning pretrained models #33

Open ANUSHAJAY opened 2 years ago

ANUSHAJAY commented 2 years ago

Hello, I must thank you for sharing your good work with us. I however have one question about the implementation. I am trying to perform pruning on a pretrained model. In one of your scripts it has been given as: python pruning_train_from_pretrain.py $DOME_HOME/datasets/ILSVRC2012 -a resnet18 --save_dir ./snapshots/Pretrain-resnet18-rate-0.7 --rate 0.7 --layer_begin 0 --layer_end 57 --layer_inter 3 --workers 36 --use_pretrain --lr 0.01 However, there are a couple of problems here: error: unrecognized arguments: --use_pretrain --lr 0.01 Also, I don't understand how to incorporate the "use_pretrain" in my code. I am also unable to find "pruning_train_from_pretrain.py" in your repository. I would be much obliged if you could help me with this. Keep up the good work! Thanks in advance