forked from dmlc/cxxnet
-
Notifications
You must be signed in to change notification settings - Fork 0
/
MNIST_CONV.conf
60 lines (57 loc) · 1.13 KB
/
MNIST_CONV.conf
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
# example configure file for mnist
# training iterator
data = train
iter = mnist
path_img = "./data/train-images-idx3-ubyte.gz"
path_label = "./data/train-labels-idx1-ubyte.gz"
input_flat = 0
shuffle = 1
iter = end
# evaluation iterator
eval = test
iter = mnist
input_flat = 0
path_img = "./data/t10k-images-idx3-ubyte.gz"
path_label = "./data/t10k-labels-idx1-ubyte.gz"
iter = end
netconfig=start
layer[0->1] = conv:cv1
kernel_size = 3
pad = 1
stride = 2
nchannel = 32
random_type = xavier
no_bias=0
layer[1->2] = max_pooling
kernel_size = 3
stride = 2
layer[2->3] = flatten
layer[3->3] = dropout
threshold = 0.5
layer[3->4] = fullc:fc1
nhidden = 100
init_sigma = 0.01
layer[4->5] = sigmoid:se1
layer[5->6] = fullc:fc2
nhidden = 10
init_sigma = 0.01
layer[6->6] = softmax
netconfig=end
# input shape not including batch
input_shape = 1,28,28
batch_size = 100
## global parameters
dev = gpu
save_model = 15
max_round = 15
num_round = 15
train_eval = 1
random_type = gaussian
## learning parameters
eta = 0.1
momentum = 0.9
wd = 0.0
# evaluation metric
metric = error
eval_train = 1
# end of config