forked from epfl-dlab/descartes
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathfull_pipeline_script.sh
executable file
·26 lines (26 loc) · 1.14 KB
/
full_pipeline_script.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
python ./scripts/extract_data.py \
--languages en_XX,cs_CZ,fr_XX,it_IT,fi_FI,de_DE,nl_XX,ro_RO,ru_RU,ja_XX,ar_AR,tr_TR,lt_LT,lv_LV,es_XX,et_EE,ko_KR,gu_IN,kk_KZ,my_MM,si_LK,vi_VN,zh_CN,ne_NP,hi_IN \
"$@"
python ./transformers/examples/seq2seq/run_summarization.py \
--learning_rate=3e-5 \
--do_train \
--do_eval \
--evaluation_strategy="steps" \
--freeze_encoder \
--freeze_embeds \
--max_source_length 512 \
--per_device_train_batch_size=1 \
--per_device_eval_batch_size=1 \
--num_train_epochs 5 \
--dataloader_num_workers 1 \
--tgt_lang en_XX \
--src_lang en_XX \
--save_steps 100000 \
--eval_steps 100000 \
--tokenizer_name facebook/mbart-large-cc25 \
--model_name_or_path facebook/mbart-large-cc25 \
--languages en_XX,cs_CZ,fr_XX,it_IT,fi_FI,de_DE,nl_XX,ro_RO,ru_RU,ja_XX,ar_AR,tr_TR,lt_LT,lv_LV,es_XX,et_EE,ko_KR,gu_IN,kk_KZ,my_MM,si_LK,vi_VN,zh_CN,ne_NP,hi_IN \
"$@"
python ./scripts/test_model.py \
--languages en_XX,cs_CZ,fr_XX,it_IT,fi_FI,de_DE,nl_XX,ro_RO,ru_RU,ja_XX,ar_AR,tr_TR,lt_LT,lv_LV,es_XX,et_EE,ko_KR,gu_IN,kk_KZ,my_MM,si_LK,vi_VN,zh_CN,ne_NP,hi_IN \
"$@"