diff --git a/examples/agp-pruning/alexnet.schedule_agp.yaml b/examples/agp-pruning/alexnet.schedule_agp.yaml index 3d73fd001fa612f2a9af957f63ff4b6a4e877818..1b161427e17d51e24bc102c3bf353a4fe1216ae0 100755 --- a/examples/agp-pruning/alexnet.schedule_agp.yaml +++ b/examples/agp-pruning/alexnet.schedule_agp.yaml @@ -1,6 +1,6 @@ # This schedule performs element-wise (fine grain) pruning, following the Automated Gradual Pruner (Zhu-Gupta) schedule. # -# time python3 prune_net.py -a=alexnet --lr=0.005 -p=50 ../../../data.imagenet -j 22 --epochs 90 --pretrained --compress=../imagenet/alexnet/alexnet.schedule_agp.yaml +# time python3 compress_classifier.py -a=alexnet --lr=0.005 -p=50 ../../../data.imagenet -j 22 --epochs 90 --pretrained --compress=../agp-pruning/alexnet.schedule_agp.yaml # # Parameters: #