File size: 568 Bytes
6931c7b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 |
#!/bin/sh
set -x
set -e
export OMP_NUM_THREADS=4
export MKL_NUM_THREADS=4
export NUMEXPR_NUM_THREADS=4
export KMP_INIT_AT_FORK=FALSE
PYTHON=python3
exp_name=$1
config=$2
dataset=$3
stage=$4
if [ "${stage}" == "s1" ]; then
TEST_CODE=test_multi_vq.py
else
TEST_CODE=test_multi_pred.py
fi
exp_dir=RUN/${dataset}/${exp_name}
model_dir=${exp_dir}/model
result_dir=${exp_dir}/result
mkdir -p ${model_dir} ${result_dir}
now=$(date +"%Y%m%d_%H%M%S")
export PYTHONPATH=./
$PYTHON -u main/${TEST_CODE} \
--config=${config} \
2>&1 | tee ${exp_dir}/test-$now.log
|