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

test_iter (should be 100 instead of 50)

1 files changed, 21 insertions, 1 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"