Spaces:
Running
on
T4
Running
on
T4
File size: 419 Bytes
d4b77ac |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
if [ -z "$3" ]
then
echo "Need input parameter!"
echo "Usage: bash `basename "$0"` \$MODEL \$DATASET \$GPUID"
exit
fi
ROOT=`git rev-parse --show-toplevel`
export PYTHONPATH=$ROOT:$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 \
--mask \
--dataset $dataset 2>&1 | tee logs/test.log
|