-
Notifications
You must be signed in to change notification settings - Fork 6
/
Copy pathtrain_base.sh
4 lines (4 loc) · 1013 Bytes
/
train_base.sh
1
2
3
4
mkdir basemodule
t2t-trainer --t2t_usr_dir=./ai_data --data_dir=./t2t_data --problem=translate_enzh_sub92k --model=transformer --hparams_set=transformer_base_single_gpu --output_dir=./basemodule --train_steps=10000 --eval_steps=1000 --hparams="learning_rate=1.18" --worker_gpu=1 --worker_gpu_memory_fraction=0.90 --local_eval_frequency=1000
t2t-trainer --t2t_usr_dir=./ai_data --data_dir=./t2t_data --problem=translate_enzh_sub92k --model=transformer --hparams_set=transformer_base_single_gpu --output_dir=./basemodule --train_steps=40000 --eval_steps=1000 --hparams="learning_rate=0.68" --worker_gpu=1 --worker_gpu_memory_fraction=0.90 --local_eval_frequency=1000
t2t-trainer --t2t_usr_dir=./ai_data --data_dir=./t2t_data --problem=translate_enzh_sub92k --model=transformer --hparams_set=transformer_base_single_gpu --output_dir=./basemodule --train_steps=180000 --eval_steps=1000 --hparams="learning_rate=0.318" --worker_gpu=1 --worker_gpu_memory_fraction=0.90 --local_eval_frequency=1000