-
Notifications
You must be signed in to change notification settings - Fork 6
/
01_zero_shot_search.sh
70 lines (67 loc) · 2 KB
/
01_zero_shot_search.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
70
### for tiny search space
python -m torch.distributed.launch \
--nproc_per_node=8 \
--use_env \
two_step_search.py \
--gp \
--change_qk \
--relative_position \
--dist-eval \
--batch-size 64 \
--data-free \
--score-method left_super_taylor6 \
--block-score-method-for-head balance_taylor6_max_dim \
--block-score-method-for-mlp balance_taylor6_max_dim \
--cand-per-interval 1 \
--param-interval 1.0 \
--min_param_limits 5 \
--param_limits 12 \
--data-path ../datas/imagenet \
--cfg ./experiments/supernet/supernet-T.yaml \
--interval-cands-output ./interval_cands/tiny.json
python candidates_to_choices.py ./interval_cands/tiny.json ./experiments/supernet/tiny.yaml
### for small search space
#python -m torch.distributed.launch \
#--nproc_per_node=8 \
#--use_env \
#two_step_search.py \
#--gp \
#--change_qk \
#--relative_position \
#--dist-eval \
#--batch-size 64 \
#--data-free \
#--score-method left_super_taylor6 \
#--block-score-method-for-head balance_taylor6_max_dim \
#--block-score-method-for-mlp balance_taylor6_max_dim \
#--cand-per-interval 1 \
#--param-interval 5.0 \
#--min_param_limits 13 \
#--param_limits 33 \
#--data-path ../datas/imagenet \
#--cfg ./experiments/supernet/supernet-S.yaml \
#--interval-cands-output ./interval_cands/small.json
#
#python candidates_to_choices.py ./interval_cands/small.json ./experiments/supernet/small.yaml
### for base search space
#python -m torch.distributed.launch \
#--nproc_per_node=8 \
#--use_env two_step_search.py \
#--gp \
#--change_qk \
#--relative_position \
#--dist-eval \
#--batch-size 64 \
#--data-free \
#--score-method left_super_taylor6 \
#--block-score-method-for-head balance_taylor6_max_dim \
#--block-score-method-for-mlp balance_taylor6_max_dim \
#--cand-per-interval 1 \
#--param-interval 12.0 \
#--min_param_limits 30 \
#--param_limits 70 \
#--data-path ../datas/imagenet \
#--cfg ./experiments/supernet/supernet-B.yaml \
#--interval-cands-output ./interval_cands/base.json
#
#python candidates_to_choices.py ./interval_cands/base.json ./experiments/supernet/base.yaml