Merge pull request #35 from allenai/fix_reg

fix the script for running regular t0 training
allenai · Aug 7, 2022 · 830f411 · 830f411
2 parents f599682 + 2f1da98
commit 830f411
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/t0_reg_train.sh b/scripts/t0_reg_train.sh
@@ -10,7 +10,7 @@ HF_DATASETS_OFFLINE=1 TRANSFORMERS_OFFLINE=1 python3 -m t5x.train \
   --gin_search_paths=gins \
   --gin_file="hyper_xl.gin" \
   --gin_file="t0_train.gin" \
-  --gin_file=hyper_network.HyperT5Config.add_adapters=False \
+  --gin.hyper_network.HyperT5Config.add_adapters=False \
   --gin.MODEL_DIR=\"${MODEL_DIR}\" \
   --gin.TRAIN_STEPS=1212200 \
   --gin.partitioning.PjitPartitioner.num_partitions=8 \