From 856c428fe0f1f451f8a22acee50c2e3af33c834e Mon Sep 17 00:00:00 2001 From: Jacek Czaja Date: Wed, 3 Jun 2020 15:13:55 +0200 Subject: [PATCH] Makeing script more flexible (#4681) Added missing vars Author: Jacek Czaja --- PaddleNLP/sentiment_classification/run.sh | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) mode change 100644 => 100755 PaddleNLP/sentiment_classification/run.sh diff --git a/PaddleNLP/sentiment_classification/run.sh b/PaddleNLP/sentiment_classification/run.sh old mode 100644 new mode 100755 index 63e54f22..8561d7c5 --- a/PaddleNLP/sentiment_classification/run.sh +++ b/PaddleNLP/sentiment_classification/run.sh @@ -9,15 +9,18 @@ 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() { python -u run_classifier.py \ --task_name ${TASK_NAME} \ - --use_cuda true \ + --use_cuda $USE_CUDA \ --do_train true \ --do_val false \ --do_infer false \ + --model_type $MODEL_TYPE \ --batch_size 8 \ --data_dir ${DATA_PATH} \ --vocab_path ${DATA_PATH}/word_dict.txt \ @@ -33,10 +36,11 @@ train() { evaluate() { python -u run_classifier.py \ --task_name ${TASK_NAME} \ - --use_cuda true \ + --use_cuda $USE_CUDA \ --do_train false \ --do_val true \ --do_infer false \ + --model_type $MODEL_TYPE \ --batch_size 10 \ --data_dir ${DATA_PATH} \ --vocab_path ${DATA_PATH}/word_dict.txt \ @@ -48,10 +52,11 @@ evaluate() { infer() { python -u run_classifier.py \ --task_name ${TASK_NAME} \ - --use_cuda true \ + --use_cuda $USE_CUDA \ --do_train false \ --do_val false \ --do_infer true \ + --model_type $MODEL_TYPE \ --batch_size 10 \ --data_dir ${DATA_PATH} \ --vocab_path ${DATA_PATH}/word_dict.txt \ -- GitLab