export FLAGS_eager_delete_tensor_gb=0.0
export CUDA_VISIBLE_DEVICES=0

# User can select chnsenticorp, nlpcc_dbqa, lcqmc and so on for different task
# Support ChnSentiCorp	NLPCC_DBQA	LCQMC	MRPC	QQP	SST-2
#         CoLA	QNLI	RTE	MNLI (or MNLI_m) 	MNLI_mm）	XNLI
# for XNLI: Specify the language with an underscore like xnli_zh.
#       ar: Arabic      bg: Bulgarian      de: German
#       el: Greek       en: English        es: Spanish
#       fr: French      hi: Hindi          ru: Russian
#       sw: Swahili     th: Thai           tr: Turkish
#       ur: Urdu        vi: Vietnamese     zh: Chinese (Simplified)
DATASET="ChnSentiCorp"
CKPT_DIR="./ckpt_${DATASET}"

python -u predict.py --checkpoint_dir=$CKPT_DIR \
                            --max_seq_len=128 \
                            --use_gpu=True \
                            --dataset=${DATASET} \
                            --batch_size=150 \