From 5fdb0573f4c9de00007c345e4db6230025b2b9b5 Mon Sep 17 00:00:00 2001 From: Neta Zmora <neta.zmora@intel.com> Date: Tue, 8 May 2018 19:36:45 +0300 Subject: [PATCH] imagenet baseline schedule: fix paths Fix the path to the example schedule for ImageNet baseline training, and to the ImageNet dataset --- examples/agp-pruning/mobilenet_imagenet_baseline_training.yaml | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/examples/agp-pruning/mobilenet_imagenet_baseline_training.yaml b/examples/agp-pruning/mobilenet_imagenet_baseline_training.yaml index 9b834f6..f126de6 100755 --- a/examples/agp-pruning/mobilenet_imagenet_baseline_training.yaml +++ b/examples/agp-pruning/mobilenet_imagenet_baseline_training.yaml @@ -15,8 +15,7 @@ # snapshot: 5000 # snapshot_prefix: "mobilenet" -# python compress_classifier.py -a mobilenet_050 --compress ../mobilenet/mobilenet_imagenet_baseline_training.yaml -j 22 ~/datasets/imagenet -p 50 -b 256 --epochs 120 --lr 0.1 --wd 0.0001 --momentum 0.9 - +# python compress_classifier.py -a mobilenet_050 --compress ../mobilenet/mobilenet_imagenet_baseline_training.yaml -j 22 ../../../data.imagenet -p 50 -b 256 --epochs 120 --lr 0.1 --wd 0.0001 --momentum 0.9 lr_schedulers: training_lr: class: PolynomialLR -- GitLab