Added missing vars

PaddlePaddle · Jun 2, 2020 · 7cf5854 · 7cf5854
1 parent 863afeb
commit 7cf5854
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/PaddleNLP/sentiment_classification/run.sh b/PaddleNLP/sentiment_classification/run.sh
@@ -9,6 +9,8 @@ TASK_NAME='senta'
 DATA_PATH=./senta_data/
 CKPT_PATH=./save_models
 MODEL_PATH=./save_models/step_1800/
+MODEL_TYPE=${MODEL_TYPE:-bilstm_net}
+USE_CUDA=${USE_CUDA:-true}
 
 # run_train on train.tsv and do_val on test.tsv
 train() {
@@ -18,7 +20,7 @@ train() {
         --do_train true \
         --do_val false \
         --do_infer false \
-        --model_type ${MODEL_TYPE} \
+        --model_type $MODEL_TYPE \
         --batch_size 8 \
         --data_dir ${DATA_PATH} \
         --vocab_path ${DATA_PATH}/word_dict.txt \