-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathrun_srprs.sh
43 lines (37 loc) · 1.04 KB
/
run_srprs.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
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
cd src
version="TEA"
gpus='1'
# en_de
paras=""
paras="$paras --datasets_root ../data"
paras="$paras --dataset en_de_15k_V1"
paras="$paras --result_root ../outputs"
paras="$paras --pretrain_bert_path ../pre_trained_models/bert-base-multilingual-uncased"
paras="$paras --log"
paras="$paras --relation"
paras="$paras --neighbor"
paras="$paras --nsp"
paras="$paras --template_id 4"
paras="$paras --version ${version}"
echo $paras
python -u TEAPreprocess.py $paras
paras="$paras --fold 1"
paras="$paras --gpus ${gpus}"
python -u TEATrain.py $paras
# en_fr
paras=""
paras="$paras --datasets_root ../data"
paras="$paras --dataset en_de_15k_V1"
paras="$paras --result_root ../outputs"
paras="$paras --pretrain_bert_path ../pre_trained_models/bert-base-multilingual-uncased"
paras="$paras --log"
paras="$paras --relation"
paras="$paras --neighbor"
paras="$paras --nsp"
paras="$paras --template_id 4"
paras="$paras --version ${version}"
echo $paras
python -u TEAPreprocess.py $paras
paras="$paras --fold 1"
paras="$paras --gpus ${gpus}"
python -u TEATrain.py $paras