From 04195be9f1aea3dbd57979f823446e28f0a80c9d Mon Sep 17 00:00:00 2001 From: Neta Zmora <neta.zmora@intel.com> Date: Mon, 7 May 2018 15:05:17 +0300 Subject: [PATCH] Fix Alexnet AGP sample command-line string --- examples/agp-pruning/alexnet.schedule_agp.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/agp-pruning/alexnet.schedule_agp.yaml b/examples/agp-pruning/alexnet.schedule_agp.yaml index 3d73fd0..1b16142 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: # -- GitLab