未验证 提交 e1e4b763 编写于 作者: J Jacek Czaja 提交者: GitHub

Fix to sentiment classification script (#4682)

上级 6a0d4500
...@@ -21,7 +21,7 @@ train() { ...@@ -21,7 +21,7 @@ train() {
--do_train true \ --do_train true \
--do_val false \ --do_val false \
--do_infer false \ --do_infer false \
--model_type ${MODEL_TYPE} \ --model_type $MODEL_TYPE \
--batch_size 8 \ --batch_size 8 \
--data_dir ${DATA_PATH} \ --data_dir ${DATA_PATH} \
--vocab_path ${DATA_PATH}/word_dict.txt \ --vocab_path ${DATA_PATH}/word_dict.txt \
...@@ -41,7 +41,7 @@ evaluate() { ...@@ -41,7 +41,7 @@ evaluate() {
--do_train false \ --do_train false \
--do_val true \ --do_val true \
--do_infer false \ --do_infer false \
--model_type ${MODEL_TYPE} \ --model_type $MODEL_TYPE \
--batch_size 10 \ --batch_size 10 \
--data_dir ${DATA_PATH} \ --data_dir ${DATA_PATH} \
--vocab_path ${DATA_PATH}/word_dict.txt \ --vocab_path ${DATA_PATH}/word_dict.txt \
...@@ -57,7 +57,7 @@ infer() { ...@@ -57,7 +57,7 @@ infer() {
--do_train false \ --do_train false \
--do_val false \ --do_val false \
--do_infer true \ --do_infer true \
--model_type ${MODEL_TYPE} \ --model_type $MODEL_TYPE \
--batch_size 10 \ --batch_size 10 \
--data_dir ${DATA_PATH} \ --data_dir ${DATA_PATH} \
--vocab_path ${DATA_PATH}/word_dict.txt \ --vocab_path ${DATA_PATH}/word_dict.txt \
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册