Update train.sh

zhongwanjun · May 27, 2022 · 5ca270b · 5ca270b
1 parent d1a3fd9
commit 5ca270b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/corpus_construction/question_generation/train.sh b/corpus_construction/question_generation/train.sh
@@ -6,8 +6,8 @@ export BASIC_DIR=../../../
 for DATA_NAME in race_negopt_withq
 do
 python -m torch.distributed.launch --nproc_per_node=8 run_qg.py \
-    --model_name_or_path t5-large \
-    --output_dir ${BASIC_DIR}/qg_model/t5-large-qg-${DATA_NAME} \
+    --model_name_or_path google/t5-v1_1-base \
+    --output_dir ${BASIC_DIR}/qg_model/t5-base-qg-${DATA_NAME} \
     --data_dir ${BASIC_DIR}/qg_data/${DATA_NAME}/ \
     --train_file_path train.jsonl \
     --valid_file_path validation.jsonl \