diff options
4 files changed, 39 insertions, 10 deletions
diff --git a/examples/mnist/mnist_autoencoder.prototxt b/examples/mnist/mnist_autoencoder.prototxt index b2bce47b..45d08023 100644 --- a/examples/mnist/mnist_autoencoder.prototxt +++ b/examples/mnist/mnist_autoencoder.prototxt @@ -18,6 +18,23 @@ layers { name: "data" type: DATA data_param { + source: "examples/mnist/mnist_train_lmdb" + backend: LMDB + batch_size: 100 + transform_param { + scale: 0.0039215684 + } + } + include: { + phase: TEST + stage: 'test-on-train' + } +} +layers { + top: "data" + name: "data" + type: DATA + data_param { source: "examples/mnist/mnist_test_lmdb" backend: LMDB batch_size: 100 @@ -25,7 +42,10 @@ layers { scale: 0.0039215684 } } - include: { phase: TEST } + include: { + phase: TEST + stage: 'test-on-test' + } } layers { bottom: "data" diff --git a/examples/mnist/mnist_autoencoder_solver.prototxt b/examples/mnist/mnist_autoencoder_solver.prototxt index af1202fc..be0939d9 100644 --- a/examples/mnist/mnist_autoencoder_solver.prototxt +++ b/examples/mnist/mnist_autoencoder_solver.prototxt @@ -1,6 +1,9 @@ net: "examples/mnist/mnist_autoencoder.prototxt" -test_iter: 50 -test_interval: 100 +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.0001 lr_policy: "fixed" diff --git a/examples/mnist/mnist_autoencoder_solver_adagrad.prototxt b/examples/mnist/mnist_autoencoder_solver_adagrad.prototxt index fa7d65cd..641ce8a0 100644 --- a/examples/mnist/mnist_autoencoder_solver_adagrad.prototxt +++ b/examples/mnist/mnist_autoencoder_solver_adagrad.prototxt @@ -1,6 +1,9 @@ -net: "mnist_autoencoder.prototxt" -test_iter: 50 -test_interval: 100 +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" diff --git a/examples/mnist/mnist_autoencoder_solver_nesterov.prototxt b/examples/mnist/mnist_autoencoder_solver_nesterov.prototxt index 17487301..254dceec 100644 --- a/examples/mnist/mnist_autoencoder_solver_nesterov.prototxt +++ b/examples/mnist/mnist_autoencoder_solver_nesterov.prototxt @@ -1,6 +1,9 @@ -net: "mnist_autoencoder.prototxt" -test_iter: 50 -test_interval: 100 +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.0001 lr_policy: "fixed" @@ -8,7 +11,7 @@ display: 20 max_iter: 4000000 weight_decay: 0.0005 snapshot: 10000 -snapshot_prefix: "mnist_autoencoder_train" +snapshot_prefix: "examples/mnist/mnist_autoencoder_nesterov_train" momentum: 0.95 # solver mode: CPU or GPU solver_mode: GPU |