export FLAGS_eager_delete_tensor_gb=0.0
export CUDA_VISIBLE_DEVICES=0

CKPT_DIR="./ckpt_stsb"

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