predict.sh

set -eux

export CUDA_VISIBLE_DEVICES=0 
export BATCH_SIZE=64
export CKPT=./checkpoints/model_10000.pdparams
export DATASET_FILE=DuEE_DuIE_data/data_DuIE/test_data.json

python run_duie.py \
    --data_path DuEE_DuIE_data/data_DuIE/ \
    --do_predict \
    --init_checkpoint $CKPT \
    --predict_data_file $DATASET_FILE \
    --max_seq_length 128 \
    --batch_size $BATCH_SIZE