-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathrun_fb.sh
69 lines (59 loc) · 1.44 KB
/
run_fb.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
60
61
62
63
64
65
66
67
68
69
#!/usr/bin/env bash
source ~/anaconda3/etc/profile.d/conda.sh
conda activate your_env_name
model_name=('PMF')
data_choice=('FBYG15K' 'FBDB15K')
data_split=('norm')
data_rate=(0.2)
use_surface=(0) # whether to use surface information
cross_modal=(1) # whether to apply cross-modal loss
freeze=(1) # whether to apply PMF
il=(1) # whether to apply iterative learning
for cm in "${cross_modal[@]}"
do
for mn in "${model_name[@]}"
do
for dc in "${data_choice[@]}"
do
for ds in "${data_split[@]}"
do
for dr in "${data_rate[@]}"
do
for us in "${use_surface[@]}"
do
for de in "${freeze[@]}"
do
echo "model: " "${mn}"
echo "data_choice: " "${dc}"
echo "data_split: " "${ds}"
echo "data_rate: " "${dr}"
echo "use_surface: " "${us}"
echo "cross_modal: " "${cm}"
echo "freeze" "${de}"
CUDA_VISIBLE_DEVICES=0,1,2,3 python main.py \
--eval_epoch 1 \
--model_name "${mn}" \
--data_choice "${dc}" \
--data_split "${ds}" \
--data_rate "${dr}"\
--semi_learn_step 5\
--il "${il}" \
--lr 0.001 \
--save_model 1 \
--csls \
--csls_k 3 \
--exp_name PMF \
--exp_id ${dc}_${ds}_il${il}_sf${us} \
--use_surface ${us} \
--cross_modal "${cm}" \
--freeze "${de}" \
--enable_sota
sleep 10
done
done
done
done
done
done
done
done