-
Notifications
You must be signed in to change notification settings - Fork 0
/
search_EA.sh
41 lines (39 loc) · 1.54 KB
/
search_EA.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
#!/bin/sh
:<<Experiment_NOTE
[Experiment NOTE]
-This area will not execute which only used to record the experiment description.
EA with less mutation and more crossover. And mutation operator V3 will not change the first bit of each unit.
Experiment_NOTE
MODE=EXPERIMENT
DEVICE=cuda:3
python3 search_EA.py --seed=0 \
--save_root=./Experiments/ \
--generations=25 \
--layers=1 \
--channels=16 \
--keep_prob=0.6 \
--drop_path_keep_prob=0.8 \
--classes=10 \
--pop_size=40 \
--obj_num=2 \
--mutate_rate=0.5 \
--crossover_rate=1 \
--mode=$MODE \
--data_path=./Res/Dataset/ \
--num_work=16 \
--train_batch_size=196 \
--eval_batch_size=196 \
--l2_reg=3e-4 \
--momentum=0.9 \
--lr_min=0.001 \
--lr_max=0.1 \
--epochs=20 \
--split_train_for_valid=0.8 \
--small_set=1 \
--surrogate_allowed=False \
--surrogate_path=./Res/PretrainModel/ \
--surrogate_premodel=2019_12_28_06_03_12 \
--surrogate_step=5 \
--surrogate_search_times=1000 \
--surrogate_preserve_topk=2 \
--device=$DEVICE