mirror of
https://github.com/zhunzhong07/person-re-ranking.git
synced 2025-06-03 14:59:19 +08:00
20 lines
433 B
Plaintext
20 lines
433 B
Plaintext
|
net: "examples/mnist/mnist_autoencoder.prototxt"
|
||
|
test_state: { stage: 'test-on-train' }
|
||
|
test_iter: 500
|
||
|
test_state: { stage: 'test-on-test' }
|
||
|
test_iter: 100
|
||
|
test_interval: 500
|
||
|
test_compute_loss: true
|
||
|
base_lr: 0.01
|
||
|
lr_policy: "step"
|
||
|
gamma: 0.1
|
||
|
stepsize: 10000
|
||
|
display: 100
|
||
|
max_iter: 65000
|
||
|
weight_decay: 0.0005
|
||
|
snapshot: 10000
|
||
|
snapshot_prefix: "examples/mnist/mnist_autoencoder"
|
||
|
momentum: 0.9
|
||
|
# solver mode: CPU or GPU
|
||
|
solver_mode: GPU
|