-
Notifications
You must be signed in to change notification settings - Fork 6
/
03_evaluation.sh
50 lines (48 loc) · 1.1 KB
/
03_evaluation.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
### eval PreNAS_tiny
python -m torch.distributed.launch \
--nproc_per_node=8 \
--use_env \
supernet_train.py \
--gp \
--change_qk \
--relative_position \
--mode retrain \
--dist-eval \
--batch-size 128 \
--eval \
--data-path ../datas/imagenet \
--cfg ./experiments/supernet/tiny.yaml \
--candfile ./interval_cands/tiny.json \
--resume ./output/tiny/checkpoint.pth
### eval PreNAS_small
#python -m torch.distributed.launch \
#--nproc_per_node=8 \
#--use_env \
#supernet_train.py \
#--gp \
#--change_qk \
#--relative_position \
#--mode retrain \
#--dist-eval \
#--batch-size 128 \
#--eval \
#--data-path ../datas/imagenet \
#--cfg ./experiments/supernet/small.yaml \
#--candfile ./interval_cands/small.json \
#--resume ./output/small/checkpoint.pth
### eval PreNAS_base
#python -m torch.distributed.launch \
#--nproc_per_node=8 \
#--use_env \
#supernet_train.py \
#--gp \
#--change_qk \
#--relative_position \
#--mode retrain \
#--dist-eval \
#--batch-size 128 \
#--eval \
#--data-path ../datas/imagenet \
#--cfg ./experiments/supernet/base.yaml \
#--candfile ./interval_cands/base.json \
#--resume ./output/base/checkpoint.pth