export CUDA_VISIBLE_DEVICES=0
# User can select chnsenticorp, nlpcc_dbqa, lcqmc for different task DATASET="chnsenticorp"
CKPT_DIR="./ckpt_${DATASET}" # Recommending hyper parameters for difference task # ChnSentiCorp: batch_size=24, weight_decay=0.01, num_epoch=3, max_seq_len=128, lr=5e-5 # NLPCC_DBQA: batch_size=8, weight_decay=0.01, num_epoch=3, max_seq_len=512, lr=2e-5 # LCQMC: batch_size=32, weight_decay=0, num_epoch=3, max_seq_len=128, lr=2e-5 python -u text_classifier.py \ --batch_size=24 \
--use_gpu=True \
--dataset=${DATASET} \ --checkpoint_dir=${CKPT_DIR} \ --learning_rate=5e-5 \ --weight_decay=0.01 \
--max_seq_len=128 \
--num_epoch=3 \ --use_pyreader=True \ --use_data_parallel=True \
