From 84c52bac6d1d16c727ce7120acf9d1773b291d51 Mon Sep 17 00:00:00 2001
From: Zac Wellmer <zacwellmer@gmail.com>
Date: Thu, 16 Feb 2017 20:03:22 -0800
Subject: [PATCH] Fixed script so it uses the proposed initialization network

Originally would always use imagenet. Regardless of user input
---
 experiments/scripts/faster_rcnn_end2end.sh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/experiments/scripts/faster_rcnn_end2end.sh b/experiments/scripts/faster_rcnn_end2end.sh
index be09f430..e7c13058 100755
--- a/experiments/scripts/faster_rcnn_end2end.sh
+++ b/experiments/scripts/faster_rcnn_end2end.sh
@@ -49,7 +49,7 @@ exec &> >(tee -a "$LOG")
 echo Logging output to "$LOG"
 
 time python ./tools/train_net.py --device ${DEV} --device_id ${DEV_ID} \
-  --weights data/pretrain_model/VGG_imagenet.npy \
+  --weights ${NET} \
   --imdb ${TRAIN_IMDB} \
   --iters ${ITERS} \
   --cfg experiments/cfgs/faster_rcnn_end2end.yml \