-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathtrain1-4.sh
44 lines (41 loc) · 1.32 KB
/
train1-4.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
wandb disabled
# nets=( EEGNet ShallowConvNet SCCNet TSception )
nets=( EEGNet )
# subject_list=( S01 S02 S03 S04 S05 S06 S07 S08 S09 )
subject_list=( S01 )
# method=( Single X X_finetune X_mix)
methods=( Single )
lr_list=( 1e-3 )
batch_list=( 32 )
# shuffle=( True False )
shuffle_list=( False )
for net in "${nets[@]}"
do
for subject in "${subject_list[@]}"
do
for method in "${methods[@]}"
do
for lr_rate in "${lr_list[@]}"
do
for batch in "${batch_list[@]}"
do
for shuffle in "${shuffle_list[@]}"
do
python main.py \
--name ${net}_${subject}_${method}_lr${lr_rate}_b${batch}_sh${shuffle} \
--net ${net} \
--subject ${subject} \
--method ${method} \
--learning_rate ${lr_rate} \
--batch ${batch} \
--shuffle ${shuffle} \
--epoch 100 \
--finetune_epoch 30 \
--gpu 0 \
--tags ${net} ${subject} ${method} batch_${batch} lr_${lr_rate} ${shuffle}
done
done
done
done
done
done