mirror of
https://github.com/zhunzhong07/person-re-ranking.git
synced 2025-06-03 14:59:19 +08:00
30 lines
871 B
Plaintext
30 lines
871 B
Plaintext
# The train/test net protocol buffer definition
|
|
net: "examples/mnist/lenet_train_test.prototxt"
|
|
# test_iter specifies how many forward passes the test should carry out.
|
|
# In the case of MNIST, we have test batch size 100 and 100 test iterations,
|
|
# covering the full 10,000 testing images.
|
|
test_iter: 100
|
|
# Carry out testing every 500 training iterations.
|
|
test_interval: 500
|
|
# The base learning rate, momentum and the weight decay of the network.
|
|
base_lr: 0.01
|
|
momentum: 0.9
|
|
weight_decay: 0.0005
|
|
# The learning rate policy
|
|
lr_policy: "multistep"
|
|
gamma: 0.9
|
|
stepvalue: 5000
|
|
stepvalue: 7000
|
|
stepvalue: 8000
|
|
stepvalue: 9000
|
|
stepvalue: 9500
|
|
# Display every 100 iterations
|
|
display: 100
|
|
# The maximum number of iterations
|
|
max_iter: 10000
|
|
# snapshot intermediate results
|
|
snapshot: 5000
|
|
snapshot_prefix: "examples/mnist/lenet_multistep"
|
|
# solver mode: CPU or GPU
|
|
solver_mode: GPU
|