-
Notifications
You must be signed in to change notification settings - Fork 0
/
run_rl_nosw_KK_sb.sh
37 lines (35 loc) · 1.86 KB
/
run_rl_nosw_KK_sb.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
#!/bin/sh
# _degree_ada linearly adapting the avg weights
# avg
# our
# SRPRS/en_fr data/fb_dbp
CUDA_VISIBLE_DEVICES=1 python3 ls_mab_comb_ori_Comb-decodeconcat-tunelast18-round2-direct.py --log gcnalign \
--model_name "GCNAPP_contras_active-decodeconcat-Multi-Iter-epoch3-tunelast18-round2-direct" \
--finetune_dataset "KK100-Multi-Iter-epoch3-tunelast18-round2-direct" \
--finetune_epoch 3 \
--sb_fine_tune 1 \
--load_from_ori 0 \
--seed 2020\
--data_dir "data/kkv4_prime" \
--rate 0.3 \
--epoch 300 \
--check 300 \
--update 10 \
--train_batch_size -1 \
--encoder "GCN-Align" \
--encoder1 "APP" \
--hiddens "100,100,100" \
--heads "2,2" \
--decoder "Align" \
--sampling "N" \
--k "25" \
--margin "1" \
--alpha "1" \
--feat_drop 0.0 \
--lr 0.002\
--train_dist "euclidean" \
--test_dist "euclidean" \
--mytest True \
--sbert True \
--sb_w "w1" \
--neg_scale 1