-
Notifications
You must be signed in to change notification settings - Fork 0
/
infer.sh
59 lines (55 loc) · 1.8 KB
/
infer.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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
# batchs=(32)
# epochs=(20) # 20: bert, 30: 나머지
# task=(snips)
# n_decs=(1) # 1 2
# model_types=(bert)
# for epoch in ${epochs[@]}
# do
# for batch in ${batchs[@]}
# do
# for n_dec in ${n_decs[@]}
# do
# for model_type in ${model_types[@]}
# do
# CUDA_VISIBLE_DEVICES=0 python main.py --task ${task} \
# --model_type ${model_type} \
# --model_dir ${task}_${model_type}_${n_dec}dec_${epoch}_${batch} \
# --do_eval \
# --max_seq_len 50 \
# --n_dec ${n_dec} \
# --num_train_epochs $epoch \
# --train_batch_size $batch
# done
# done
# done
# done
batchs=(32)
epochs=(40) # 1 5 10 20 30 40
task=(atis)
n_decs=(2) # 1 2
model_types=(bert)
slot_loss_coefs=(1.2) # 1.1 1.2 1.3 1.4 1.5 1.6 1.7 1.8 1.9 2.0
for epoch in ${epochs[@]}
do
for batch in ${batchs[@]}
do
for n_dec in ${n_decs[@]}
do
for model_type in ${model_types[@]}
do
for slot_loss_coef in ${slot_loss_coefs[@]}
do
CUDA_VISIBLE_DEVICES=0 python main.py --task ${task} \
--model_type ${model_type} \
--model_dir ${task}_${model_type}_${n_dec}dec_${epoch}_${batch}_slotloss_${slot_loss_coef} \
--slot_loss_coef ${slot_loss_coef} \
--do_eval \
--max_seq_len 60 \
--n_dec ${n_dec} \
--num_train_epochs $epoch \
--train_batch_size $batch
done
done
done
done
done