run_predict.sh 292 字节
Newer Older
K
kinghuin 已提交
1 2 3 4 5 6 7 8 9
export FLAGS_eager_delete_tensor_gb=0.0
export CUDA_VISIBLE_DEVICES=0

CKPT_DIR="./ckpt_chnsenticorp"

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