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 \