IBM_AI/vi-opton/opton_od_train/backup_model.sh

set -e

if [ $# -lt 1 ]; then
    echo 'Usage: ./backup_model.sh model_id'
    exit -1
fi

MODEL_ID=$1
MODEL_DIR=/mnt/sdb/models/opton/$MODEL_ID
ROOT_DIR=$(cd `dirname $0`; pwd)
TASK_DIR=$ROOT_DIR/$MODEL_ID

if  [ ! -d $TASK_DIR ]; then
    echo '[FAIL] Task model folder not exist: '$TASK_DIR
    exit -1
fi

if [ -d $MODEL_DIR ]; then
    echo '[FAIL] Backup model folder already exist: '$MODEL_DIR
    exit -1
fi

mkdir -p $MODEL_DIR
cp -rf $TASK_DIR/prototxt $MODEL_DIR
cp $TASK_DIR/output/snapshot/ZF_faster_rcnn_final.caffemodel $MODEL_DIR
cp $TASK_DIR/config.py $MODEL_DIR
touch $MODEL_DIR/__init__.py
touch $MODEL_DIR/$MODEL_ID'.tag'

echo '[SUCCESS] Model backup to '$MODEL_DIR