18 lines
423 B
Plaintext
18 lines
423 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: "fixed"
|
|
display: 100
|
|
max_iter: 65000
|
|
weight_decay: 0.0005
|
|
snapshot: 10000
|
|
snapshot_prefix: "examples/mnist/mnist_autoencoder_adagrad_train"
|
|
# solver mode: CPU or GPU
|
|
solver_mode: GPU
|
|
type: "AdaGrad"
|