oguzakif's picture
init repo
d4b77ac
if [ -z "$3" ]
then
echo "Need input parameter!"
echo "Usage: bash `basename "$0"` \$MODEL \$DATASETi \$GPUID"
exit
fi
source activate siammask
ROOT=`git rev-parse --show-toplevel`
export PYTHONPATH=$ROOT:$PYTHONPATH
export PYTHONPATH=$PWD:$PYTHONPATH
mkdir -p logs
model=$1
dataset=$2
gpu=$3
CUDA_VISIBLE_DEVICES=$gpu python -u $ROOT/tools/test.py \
--config config.json \
--resume $model \
--dataset $dataset 2>&1 | tee logs/test.log