-
Notifications
You must be signed in to change notification settings - Fork 0
/
config.yml
69 lines (57 loc) · 1.46 KB
/
config.yml
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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
# Copyright (C) 2020-2021 Mo Zhou <cdluminate@gmail.com>
# Released under the Apache-2.0 License.
# --- Dataset Meta Information ---
fashion-mnist:
#t10k-images-idx3-ubyte.gz
#t10k-labels-idx1-ubyte.gz
#train-images-idx3-ubyte.gz
#train-labels-idx1-ubyte.gz
path: ~/.torch/FashionMNIST/raw/
sop:
# Ebay_train.txt
# Ebay_test.txt
path: ~/.torch/Stanford_Online_Products/
# -- fashion --
faC_lenet:
epoch: 32
lr: 0.001
batchsize: 1000
batchsize_atk: 1000 # for white-box
faC_res18:
epoch: 32
lr: 0.001
batchsize: 1000
batchsize_atk: 1000 # for white-box
faC_c2f2:
epoch: 32
lr: 0.001
batchsize: 1000
batchsize_atk: 1000 # for white-box
faE_c2f2:
epoch: 32
lr: 0.001
batchsize: 1000
batchsize_atk: 100 # for white-box
# -- stanford online product
# http://cvgl.stanford.edu/projects/lifted_struct/
sopE_res18:
epoch: 42
lr: 0.00001
batchsize: 128 # (bs=128 requires 11GB with Titan Xp)
batchsize_atk: 64 # (bs=64, titanxp 11167MB) for white-box
sopE_res50:
epoch: 42
lr: 0.00001
# bs=128 requires 11GB*3 with Titan Xp (conjrhomb requires *6)
batchsize: 128 # (bs=64 requires 11GB*2 with Titan Xp)
batchsize_atk: 32 # for white-box
sopE_dense121:
epoch: 42
lr: 0.00001
batchsize: 128 # (bs=128 requires 6~8 Cards, bs=64 requires 3~4cards)
batchsize_atk: 64
sopE_mnas:
epoch: 42
lr: 0.00001
batchsize: 128 # (bs=64 requires 1 card, bs=128 consumes 2)
batchsize_atk: 128