diff --git a/PaddleKG/DuIE_Baseline/script/predict.sh b/PaddleKG/DuIE_Baseline/script/predict.sh
index fd21992016eecf688dba9abf14564659079555b0..9e74ea4837706524812a1d86c9c76cf5dd47aca1 100755
--- a/PaddleKG/DuIE_Baseline/script/predict.sh
+++ b/PaddleKG/DuIE_Baseline/script/predict.sh
@@ -1,6 +1,7 @@
 set -eux
 
 export TASK_DATA_PATH=./data/
+export DEV_FILE=dev.json
 export MODEL_PATH=./pretrained_model/
 export CHECKPOINT=./checkpoints/step_60000/
 export TEST_SAVE=./data/
@@ -17,8 +18,8 @@ CUDA_VISIBLE_DEVICES=7 python -u ./ernie/run_duie.py \
                    --num_labels 112 \
                    --label_map_config ${TASK_DATA_PATH}relation2label.json \
                    --spo_label_map_config ${TASK_DATA_PATH}label2relation.json \
-                   --test_set ${TASK_DATA_PATH}dev_demo.json \
-                   --test_save ${TEST_SAVE}predict_test.json \
+                   --test_set ${TASK_DATA_PATH}${DEV_FILE} \
+                   --test_save ${TEST_SAVE}predict_dev.json \
                    --vocab_path ${MODEL_PATH}vocab.txt \
                    --ernie_config_path ${MODEL_PATH}ernie_config.json \
                    --use_fp16 false \
diff --git a/PaddleKG/DuIE_Baseline/script/train.sh b/PaddleKG/DuIE_Baseline/script/train.sh
index 5df26204296806381c9c08a54b843eb1a6dc333f..95e333c0c559c95dce10b208f8552dc1b566ade7 100755
--- a/PaddleKG/DuIE_Baseline/script/train.sh
+++ b/PaddleKG/DuIE_Baseline/script/train.sh
@@ -8,6 +8,8 @@ export SAVE_STEPS=5000
 export SAVE_PATH=./
 export TASK_DATA_PATH=./data/
 export MODEL_PATH=./pretrained_model/
+export TRAIN_FILE=train.json
+export DEV_FILE=dev.json
 
 export FLAGS_sync_nccl_allreduce=1
 export PYTHONPATH=./ernie:${PYTHONPATH:-}
@@ -23,8 +25,8 @@ CUDA_VISIBLE_DEVICES=7 python -u ./ernie/run_duie.py \
                    --chunk_scheme "IOB" \
                    --label_map_config ${TASK_DATA_PATH}relation2label.json \
                    --spo_label_map_config ${TASK_DATA_PATH}label2relation.json \
-                   --train_set ${TASK_DATA_PATH}train_demo.json \
-                   --dev_set ${TASK_DATA_PATH}dev_demo.json \
+                   --train_set ${TASK_DATA_PATH}${TRAIN_FILE} \
+                   --dev_set ${TASK_DATA_PATH}${DEV_FILE} \
                    --vocab_path ${MODEL_PATH}vocab.txt \
                    --ernie_config_path ${MODEL_PATH}ernie_config.json \
                    --checkpoints ${SAVE_PATH}checkpoints \