change pretrain script

yxgeee · Mar 28, 2020 · dab826f · dab826f
1 parent 4890b2b
commit dab826f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/scripts/pretrain.sh b/scripts/pretrain.sh
@@ -12,5 +12,5 @@ fi
 
 CUDA_VISIBLE_DEVICES=0,1,2,3 \
 python examples/source_pretrain.py -ds ${SOURCE} -dt ${TARGET} -a ${ARCH} --seed ${SEED} --margin 0.0 \
-	--num-instances 4 -b 64 -j 4 --warmup-step 10 --lr 0.00035 --milestones 40 70 --iters 200 --epochs 80 --eval-step 40 \
-	--logs-dir logs/${SOURCE}TO${TARGET}/${ARCH}-pretrain-${SEED}
+	--num-instances 4 -b 64 -j 4 --warmup-step 10 --lr 0.00035 --milestones 40 70 --iters 100 --epochs 80 --eval-step 40 \
+	--logs-dir logs/${SOURCE}TO${TARGET}/${ARCH}-pretrain-${SEED}