run_predict.sh 256 字节
Newer Older
K
kinghuin 已提交
1 2 3 4 5
export FLAGS_eager_delete_tensor_gb=0.0
export CUDA_VISIBLE_DEVICES=0

CKPT_DIR="./ckpt_chnsenticorp"

S
Steffy-zxf 已提交
6 7 8 9 10
python -u predict.py \
            --checkpoint_dir=$CKPT_DIR \
            --max_seq_len=128 \
            --use_gpu=True \
            --batch_size=24