| PROJECT_DIR=${HOME}"/models/t5-summary" | |
| export PYTHONPATH=${PROJECT_DIR} | |
| INITIAL_CHECKPOINT_PATH=\"gs://nb-t5x-us-central2/exp24-t5-large-ul2-mt5voc-full/checkpoint_1500000\" | |
| TRAIN_STEPS=1550000 | |
| python3 ../../t5x/t5x/train.py \ | |
| --gin_search_paths="./" \ | |
| --gin.TRAIN_STEPS=${TRAIN_STEPS} \ | |
| --gin_file="finetune_summary_ul2_mlarge.gin" \ | |
| --gin.MIXTURE_OR_TASK_NAME=\"summary\" \ | |
| --gin.MODEL_DIR=\"gs://nb-t5x-us-central2/finetuned/summary_ul2_mlarge\" \ | |
| --gin.INITIAL_CHECKPOINT_PATH=${INITIAL_CHECKPOINT_PATH} \ | |