pere commited on
Commit
c73c577
1 Parent(s): 1a76e5e

fixed training steps

Browse files
Files changed (1) hide show
  1. batch_nynorsk_balanced_lm_base.sh +1 -1
batch_nynorsk_balanced_lm_base.sh CHANGED
@@ -1,7 +1,7 @@
1
  PROJECT_DIR=${HOME}"/models/t5-nynorsk-oversetter"
2
  export PYTHONPATH=${PROJECT_DIR}
3
  INITIAL_CHECKPOINT_PATH=\"gs://nb-t5x-us-central2/norwegian_NCC_plus_English_pluss200k_balanced_bokmaal_nynorsk_pluss100k_lm_t5x_base/checkpoint_1735000\"
4
- TRAIN_STEPS=17040000
5
 
6
  python3 ../../t5x/t5x/train.py --gin_search_paths="./" --gin.TRAIN_STEPS=${TRAIN_STEPS} --gin_file="finetune_translate_base.gin" --gin.INITIAL_CHECKPOINT_PATH=${INITIAL_CHECKPOINT_PATH} --gin.MIXTURE_OR_TASK_NAME=\"translate\" --gin.MODEL_DIR=\"gs://t5x-training/finetuned/nynorsk/nynorsk_NCC_lmtest_base_v1\" &&
7
  python3 ../../t5x/t5x/train.py --gin_search_paths="./" --gin.TRAIN_STEPS=${TRAIN_STEPS} --gin_file="finetune_translate_base.gin" --gin.INITIAL_CHECKPOINT_PATH=${INITIAL_CHECKPOINT_PATH} --gin.MIXTURE_OR_TASK_NAME=\"translate\" --gin.MODEL_DIR=\"gs://t5x-training/finetuned/nynorsk/nynorsk_NCC_lmtest_base_v2\" &&
 
1
  PROJECT_DIR=${HOME}"/models/t5-nynorsk-oversetter"
2
  export PYTHONPATH=${PROJECT_DIR}
3
  INITIAL_CHECKPOINT_PATH=\"gs://nb-t5x-us-central2/norwegian_NCC_plus_English_pluss200k_balanced_bokmaal_nynorsk_pluss100k_lm_t5x_base/checkpoint_1735000\"
4
+ TRAIN_STEPS=1704000
5
 
6
  python3 ../../t5x/t5x/train.py --gin_search_paths="./" --gin.TRAIN_STEPS=${TRAIN_STEPS} --gin_file="finetune_translate_base.gin" --gin.INITIAL_CHECKPOINT_PATH=${INITIAL_CHECKPOINT_PATH} --gin.MIXTURE_OR_TASK_NAME=\"translate\" --gin.MODEL_DIR=\"gs://t5x-training/finetuned/nynorsk/nynorsk_NCC_lmtest_base_v1\" &&
7
  python3 ../../t5x/t5x/train.py --gin_search_paths="./" --gin.TRAIN_STEPS=${TRAIN_STEPS} --gin_file="finetune_translate_base.gin" --gin.INITIAL_CHECKPOINT_PATH=${INITIAL_CHECKPOINT_PATH} --gin.MIXTURE_OR_TASK_NAME=\"translate\" --gin.MODEL_DIR=\"gs://t5x-training/finetuned/nynorsk/nynorsk_NCC_lmtest_base_v2\" &&