-
Notifications
You must be signed in to change notification settings - Fork 8
/
Copy pathrun_fewnerd.sh
30 lines (26 loc) · 1.2 KB
/
run_fewnerd.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
datasets_target=('FEW-NERD-INTRA')
n_way_k_shots=('5_1')
for _dataset_target in "${datasets_target[@]}"
do
python3 main.py --mode use_type_name \
--train True \
--dataset_source $_dataset_target \
--dataset_target $_dataset_target \
--n_way_k_shot '5_1' \
--seed 42 \
--test_stage1_only False\
--test_stage2_only False \
--filter True
for _n_way_k_shot in "${n_way_k_shots[@]}"
do
python3 main.py --mode use_type_name \
--train False \
--dataset_source $_dataset_target \
--dataset_target $_dataset_target \
--n_way_k_shot $_n_way_k_shot \
--seed 42 \
--test_stage1_only False\
--test_stage2_only False \
--filter True
done
done