From 5ca270b2ee5ef622d1ca3391081e612cc66f82d7 Mon Sep 17 00:00:00 2001
From: Wanjun Zhong <779529477@qq.com>
Date: Fri, 27 May 2022 18:58:55 +0800
Subject: [PATCH] Update train.sh

---
 corpus_construction/question_generation/train.sh | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/corpus_construction/question_generation/train.sh b/corpus_construction/question_generation/train.sh
index 66cf69f..f284991 100755
--- 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 \