feat: update EE task.

Author: Alexzhuan

.gitignore

@@ -1,5 +1,4 @@
 .idea
-test.py
 data/model_data/
 **/__pycache__
 data/output/

examples/run_ctc.sh

@@ -4,7 +4,7 @@ DATA_DIR="CBLUEDatasets"
 TASK_NAME="ctc"
 MODEL_TYPE="bert"
 MODEL_DIR="data/model_data"
-MODEL_NAME="chinese-bert-base"
+MODEL_NAME="chinese-bert-wwm-ext"
 OUTPUT_DIR="data/output"
 RESULT_OUTPUT_DIR="data/result_output"
 

examples/run_ie.sh

@@ -2,9 +2,9 @@
 DATA_DIR="CBLUEDatasets"
 
 TASK_NAME="ie"
-MODEL_TYPE="zen"
+MODEL_TYPE="bert"
 MODEL_DIR="data/model_data"
-MODEL_NAME="chinese-zen-base"
+MODEL_NAME="chinese-bert-wwm-ext"
 OUTPUT_DIR="data/output"
 RESULT_OUTPUT_DIR="data/result_output"
 
@@ -23,7 +23,7 @@ if [ $# == 0 ]; then
         --result_output_dir=${RESULT_OUTPUT_DIR} \
         --do_train \
         --max_length=${MAX_LENGTH} \
-        --train_batch_size=16 \
+        --train_batch_size=32 \
         --eval_batch_size=64 \
         --learning_rate=3e-5 \
         --epochs=7 \

examples/run_qic.sh

@@ -4,7 +4,7 @@ DATA_DIR="CBLUEDatasets"
 TASK_NAME="qic"
 MODEL_TYPE="bert"
 MODEL_DIR="data/model_data"
-MODEL_NAME="chinese-bert-base"
+MODEL_NAME="chinese-bert-wwm-ext"
 OUTPUT_DIR="data/output"
 RESULT_OUTPUT_DIR="data/result_output"
 

examples/run_qqr.sh

@@ -4,7 +4,7 @@ DATA_DIR="CBLUEDatasets"
 TASK_NAME="qqr"
 MODEL_TYPE="bert"
 MODEL_DIR="data/model_data"
-MODEL_NAME="chinese-bert-base"
+MODEL_NAME="chinese-bert-wwm-ext"
 OUTPUT_DIR="data/output"
 RESULT_OUTPUT_DIR="data/result_output"
 

examples/run_qtr.sh

@@ -4,7 +4,7 @@ DATA_DIR="CBLUEDatasets"
 TASK_NAME="qtr"
 MODEL_TYPE="bert"
 MODEL_DIR="data/model_data"
-MODEL_NAME="chinese-bert-base"
+MODEL_NAME="chinese-bert-ext"
 OUTPUT_DIR="data/output"
 RESULT_OUTPUT_DIR="data/result_output"