Add "test-on-train" stage to test accuracy on the training data; correct

test_iter (should be 100 instead of 50)

diff --git a/examples/mnist/mnist_autoencoder.prototxt b/examples/mnist/mnist_autoencoder.prototxt
index b2bce47..45d0802 100644 (file)
--- 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 af1202f..be0939d 100644 (file)
--- 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 fa7d65c..641ce8a 100644 (file)
--- 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 1748730..254dcee 100644 (file)
--- 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