run_predict.sh

export FLAGS_eager_delete_tensor_gb=0.0
export CUDA_VISIBLE_DEVICES=0

CKPT_DIR="./ckpt_rc"
RES_DIR="./result"

mkdir $RES_DIR

python -u  predict.py \
                   --batch_size=12 \
                   --use_gpu=True \
                   --checkpoint_dir=${CKPT_DIR} \
                   --learning_rate=3e-5 \
                   --weight_decay=0.01 \
                   --warmup_proportion=0.1 \
                   --num_epoch=1 \
                   --max_seq_len=384 \
                   --use_pyreader=False \
                   --use_data_parallel=False \
                   --version_2_with_negative=False \
                   --result_dir=${RES_DIR}