-
Notifications
You must be signed in to change notification settings - Fork 16
/
solver.prototxt
30 lines (30 loc) · 960 Bytes
/
solver.prototxt
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
# The train/test net protocol buffer definition
net: "/home/nd/caffe_set/caffe-moon/examples/moon_tiny/train_val.prototxt"
# To skip test at start
# test_iter specifies how many forward passes the test should carry out.
# In our case, 625*32=20K is above the total number of validation images
test_iter: 313
# To have batch_size = 64
#iter_size: 2
# Carry out testing every X training iterations (same as display).
test_interval: 2500
# The base learning rate, momentum and the weight decay of the network.
base_lr: 0.00001
momentum: 0.0
weight_decay: 1e-05
# The learning rate policy
lr_policy: "inv"
gamma: 0.0001
power: 0.75
# Display every X iterations
display: 200
average_loss: 2500
# The maximum number of iterations #50 epochs
max_iter: 1000000
# snapshot intermediate results
snapshot: 5000
snapshot_prefix: "/home/nd/caffe_set/caffe-moon/examples/moon_tiny/models/moon_tiny"
# solver mode: CPU or GPU
solver_mode: GPU
type: "RMSProp"
rms_decay: 0.98