-
Notifications
You must be signed in to change notification settings - Fork 6
/
Copy pathtrain_big.sh
4 lines (4 loc) · 1005 Bytes
/
train_big.sh
1
2
3
4
mkdir bigmodule
t2t-trainer --t2t_usr_dir=./ai_data --data_dir=./t2t_data --problem=translate_enzh_sub92k --model=transformer --hparams_set=transformer_big_single_gpu --output_dir=./bigmodule --train_steps=10000 --eval_steps=1000 --hparams="learning_rate=1.18" --worker_gpu=4 --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_big_single_gpu --output_dir=./bigmodule --train_steps=40000 --eval_steps=1000 --hparams="learning_rate=0.68" --worker_gpu=4 --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_big_single_gpu --output_dir=./bigmodule --train_steps=180000 --eval_steps=1000 --hparams="learning_rate=0.318" --worker_gpu=4 --worker_gpu_memory_fraction=0.90 --local_eval_frequency=1000