-
Notifications
You must be signed in to change notification settings - Fork 2
/
white_attack_3.sh
36 lines (35 loc) · 1.02 KB
/
white_attack_3.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
# --------- CIFAR10-vgg/resnet -----------------
arch=OMPc
# ----
# model=vgg11
# model_path='./save/CIFAR10-OMPc-vgg11-lamb-0.1-path-10-adv.pth'
# ----
# model=vgg13
# model_path='./save/CIFAR10-OMPc-vgg13-lamb-0.1-path-10-adv.pth'
# ----
model=vgg16
model_path='./save/CIFAR10-OMPc-vgg16-lamb-0.1-path-10-adv.pth'
# ----
# model=vgg19
# model_path='./save/CIFAR10-OMPc-vgg19-lamb-0.1-path-10-adv.pth'
# ----
# model=resnet20
# model_path='./save/CIFAR10-OMPc-resnet20-lamb-0.1-path-10-adv.pth'
# ----
# model=resnet32
# model_path='./save/CIFAR10-OMPc-resnet32-lamb-0.1-path-10-adv.pth'
# -------- hyper-parameters ---------------------
num_paths=10
# -------- CIFAR10 ------------------------------
dataset=CIFAR10
data_dir='/media/Disk1/KunFang/data/CIFAR10/'
# -----------------------------------------------
gpu_id=0
python white_attack_3.py \
--arch ${arch} \
--model ${model} \
--model_path ${model_path} \
--dataset ${dataset} \
--data_dir ${data_dir} \
--gpu_id ${gpu_id} \
--num_paths ${num_paths}