pere
/

t5-summary

Model card Files Files and versions Community

pere commited on Feb 22, 2023

Commit

84c7990

•

1 Parent(s): a0b6735

all

Files changed (1) hide show

summary_all_base.sh +2 -2

summary_all_base.sh CHANGED Viewed

@@ -8,7 +8,7 @@ TRAIN_STEPS=1600000
 python3 ../../t5x/t5x/train.py \
   --gin_search_paths="./" \
   --gin.TRAIN_STEPS=${TRAIN_STEPS} \
-  --gin_file="finetune_summary_all_base.gin" \
-  --gin.MIXTURE_OR_TASK_NAME=\"summary\" \
   --gin.MODEL_DIR=\"gs://nb-t5x-us-central2/finetuned/summary_all_NCC_base\" \
   --gin.INITIAL_CHECKPOINT_PATH=${INITIAL_CHECKPOINT_PATH} \

 python3 ../../t5x/t5x/train.py \
   --gin_search_paths="./" \
   --gin.TRAIN_STEPS=${TRAIN_STEPS} \
+  --gin_file="finetune_summary_base.gin" \
+  --gin.MIXTURE_OR_TASK_NAME=\"summary_all\" \
   --gin.MODEL_DIR=\"gs://nb-t5x-us-central2/finetuned/summary_all_NCC_base\" \
   --gin.INITIAL_CHECKPOINT_PATH=${INITIAL_CHECKPOINT_PATH} \