From 7f82cc0cfe7e711dd1d58e520d0200a6289ad22a Mon Sep 17 00:00:00 2001
From: zqwerty <zhuq96@hotmail.com>
Date: Tue, 28 Dec 2021 01:58:07 +0000
Subject: [PATCH] update run_rg.sh: add create_data script
---
convlab2/base_models/t5/run_rg.sh | 2 ++
1 file changed, 2 insertions(+)
diff --git a/convlab2/base_models/t5/run_rg.sh b/convlab2/base_models/t5/run_rg.sh
index 60c88cfb..7c72b9c8 100644
--- a/convlab2/base_models/t5/run_rg.sh
+++ b/convlab2/base_models/t5/run_rg.sh
@@ -18,6 +18,8 @@ gradient_accumulation_steps=1
lr=1e-3
num_train_epochs=5
+python create_data.py --tasks ${task_name} --datasets ${dataset_name} --save_dir "data"
+
python -m torch.distributed.launch \
--nproc_per_node ${n_gpus} run_seq2seq.py \
--task_name ${task_name} \
--
GitLab