run_predict.sh 565 字节
Newer Older
K
kinghuin 已提交
1 2
export FLAGS_eager_delete_tensor_gb=0.0

K
kinghuin 已提交
3 4
CKPT_DIR="./ckpt_cmrc2018"
dataset=cmrc2018
K
kinghuin 已提交
5 6

python -u  predict.py \
K
kinghuin 已提交
7
                   --batch_size=8 \
K
kinghuin 已提交
8
                   --use_gpu=True \
K
kinghuin 已提交
9
                   --dataset=${dataset} \
K
kinghuin 已提交
10
                   --checkpoint_dir=${CKPT_DIR} \
K
kinghuin 已提交
11
                   --learning_rate=2.5e-5 \
K
kinghuin 已提交
12 13 14
                   --weight_decay=0.01 \
                   --warmup_proportion=0.1 \
                   --num_epoch=1 \
K
kinghuin 已提交
15
                   --max_seq_len=512 \
K
kinghuin 已提交
16
                   --use_pyreader=False \
K
kinghuin 已提交
17
                   --use_data_parallel=False