Skip to content

Commit

Permalink
fix gin config in regular t0 training.
Browse files Browse the repository at this point in the history
  • Loading branch information
yizhongw committed Aug 7, 2022
1 parent f599682 commit 2f1da98
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion scripts/t0_reg_train.sh
Original file line number Diff line number Diff line change
Expand Up @@ -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 \
Expand Down

0 comments on commit 2f1da98

Please sign in to comment.