File size: 381 Bytes
d4b77ac |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
if [ -z "$1" ]
then
echo "Need input base model!"
echo "Usage: bash `basename "$0"` \$BASE_MODEL"
exit
fi
ROOT=`git rev-parse --show-toplevel`
export PYTHONPATH=$ROOT:$PYTHONPATH
mkdir -p logs
base=$1
python -u $ROOT/tools/train_siammask_refine.py \
--config=config.json -b 64 \
-j 20 --pretrained $base \
--epochs 20 \
2>&1 | tee logs/train.log
|