d0edc0f0c874530d03ca36721c9026b254dc5b6f
[jacinto-ai/caffe-jacinto.git] / data / lenet_solver.prototxt
1 # The training protocol buffer definition
2 train_net: "lenet.prototxt"
3 # The testing protocol buffer definition
4 test_net: "lenet_test.prototxt"
5 # test_iter specifies how many forward passes the test should carry out.
6 # In the case of MNIST, we have test batch size 100 and 100 test iterations,
7 # covering the full 10,000 testing images.
8 test_iter: 100
9 # Carry out testing every 500 training iterations.
10 test_interval: 500
11 # The base learning rate, momentum and the weight decay of the network.
12 base_lr: 0.01
13 momentum: 0.9
14 weight_decay: 0.0005
15 # The learning rate policy
16 lr_policy: "inv"
17 gamma: 0.0001
18 power: 0.75
19 # Display every 100 iterations
20 display: 100
21 # The maximum number of iterations
22 max_iter: 10000
23 # snapshot intermediate results
24 snapshot: 5000
25 snapshot_prefix: "lenet"
26 # solver mode: 0 for CPU and 1 for GPU
27 solver_mode: 1