oguzakif's picture
init repo
d4b77ac
raw
history blame
381 Bytes
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