| 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 | |