run_predict.sh 909 字节
Newer Older
1
export FLAGS_eager_delete_tensor_gb=0.0
W
wuzewu 已提交
2
export CUDA_VISIBLE_DEVICES=0
Z
Zeyu Chen 已提交
3

K
kinghuin 已提交
4 5
# User can select chnsenticorp, nlpcc_dbqa, lcqmc and so on for different task
# Support ChnSentiCorp	NLPCC_DBQA	LCQMC	MRPC	QQP	SST-2
K
kinghuin 已提交
6
#         CoLA	QNLI	RTE	MNLI (or MNLI_m) 	MNLI_mm)	XNLI
K
kinghuin 已提交
7 8 9 10 11 12
# 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)
K
kinghuin 已提交
13 14
DATASET="ChnSentiCorp"
CKPT_DIR="./ckpt_${DATASET}"
K
kinghuin 已提交
15

K
kinghuin 已提交
16 17 18 19
python -u predict.py --checkpoint_dir=$CKPT_DIR \
                            --max_seq_len=128 \
                            --use_gpu=True \
                            --dataset=${DATASET} \
K
kinghuin 已提交
20
                            --batch_size=32 \