From 958b361f71295c585cc5929377cf1c100601fade Mon Sep 17 00:00:00 2001 From: Neta Zmora <neta.zmora@intel.com> Date: Sat, 28 Apr 2018 20:55:13 +0300 Subject: [PATCH] SSL 4D schedule: fix path to dataset and the number of epochs --- examples/ssl/ssl_4D-removal_finetuning.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/ssl/ssl_4D-removal_finetuning.yaml b/examples/ssl/ssl_4D-removal_finetuning.yaml index bf3f99a..b9d241b 100755 --- a/examples/ssl/ssl_4D-removal_finetuning.yaml +++ b/examples/ssl/ssl_4D-removal_finetuning.yaml @@ -1,6 +1,6 @@ # We used this schedule to fine-tune a CIFAR0-ResNet20 model which had soem of its layers regularized-away with SSL (see ssl_4D-removal_training.yaml). # -# time python3 compress_classifier.py --arch resnet20_cifar ../data.cifar10 -p=50 --lr=0.1 --epochs=250 --resume=../ssl/checkpoints/checkpoint_trained_4D_regularized_5Lremoved.pth.tar --compress=../ssl/ssl_4D-removal_finetuning.yaml -j=1 --deterministic +# time python3 compress_classifier.py --arch resnet20_cifar ../../../data.cifar10 -p=50 --lr=0.1 --epochs=70 --resume=../ssl/checkpoints/checkpoint_trained_4D_regularized_5Lremoved.pth.tar --compress=../ssl/ssl_4D-removal_finetuning.yaml -j=1 --deterministic # # Parameters: # -- GitLab