forked from fanghenshaometeor/Orthogonal-Multi-Path
-
Notifications
You must be signed in to change notification settings - Fork 0
/
white_attack_2.sh
35 lines (34 loc) · 969 Bytes
/
white_attack_2.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
# --------- CIFAR10-vgg/resnet -----------------
# --------- OMP-a
# arch=OMPa
# ----
# model=vgg16
# model_path='./save/CIFAR10-OMPa-vgg16-lamb-0.1-path-10-adv.pth'
# ----
# model=resnet20
# model_path='./save/CIFAR10-OMPa-resnet20-lamb-0.1-path-10-adv.pth'
# --------- OMP-b
arch=OMPb
# ----
# model=vgg16
# model_path='./save/CIFAR10-OMPb-vgg16-lamb-0.1-path-10-adv.pth'
# ----
model=resnet20
model_path='./save/CIFAR10-OMPb-resnet20-lamb-0.1-path-10-adv.pth'
# -------- hyper-parameters ---------------------
num_paths=10
selected_path=5
# -------- CIFAR10 ------------------------------
dataset=CIFAR10
data_dir='/media/Disk1/KunFang/data/CIFAR10/'
# -----------------------------------------------
gpu_id=0
python white_attack_2.py \
--arch ${arch} \
--model ${model} \
--model_path ${model_path} \
--dataset ${dataset} \
--data_dir ${data_dir} \
--gpu_id ${gpu_id} \
--num_paths ${num_paths} \
--selected_path ${selected_path}