make solver able to compute and display test loss

2 files changed, 1 insertions, 20 deletions

diff --git a/examples/mnist/mnist_autoencoder_solver.prototxt b/examples/mnist/mnist_autoencoder_solver.prototxt
index b11b2c46..dd4f0fec 100644
--- a/examples/mnist/mnist_autoencoder_solver.prototxt
+++ b/examples/mnist/mnist_autoencoder_solver.prototxt
@@ -2,6 +2,7 @@ train_net: "mnist_autoencoder_train.prototxt"
 test_net: "mnist_autoencoder_test.prototxt"
 test_iter: 50
 test_interval: 100
+test_compute_loss: true
 base_lr: 0.0001
 lr_policy: "fixed"
 display: 20
diff --git a/examples/mnist/mnist_autoencoder_test.prototxt b/examples/mnist/mnist_autoencoder_test.prototxt
index c106b47f..5090e82f 100644
--- a/examples/mnist/mnist_autoencoder_test.prototxt
+++ b/examples/mnist/mnist_autoencoder_test.prototxt
@@ -20,10 +20,6 @@ layers {
   top: "encode1"
   name: "encode1"
   type: INNER_PRODUCT
-  blobs_lr: 1
-  blobs_lr: 1
-  weight_decay: 1
-  weight_decay: 0
   inner_product_param {
     num_output: 1000
   }
@@ -39,10 +35,6 @@ layers {
   top: "encode2"
   name: "encode2"
   type: INNER_PRODUCT
-  blobs_lr: 1
-  blobs_lr: 1
-  weight_decay: 1
-  weight_decay: 0
   inner_product_param {
     num_output: 500
   }
@@ -58,10 +50,6 @@ layers {
   top: "encode3"
   name: "encode3"
   type: INNER_PRODUCT
-  blobs_lr: 1
-  blobs_lr: 1
-  weight_decay: 1
-  weight_decay: 0
   inner_product_param {
     num_output: 250
   }
@@ -109,10 +97,6 @@ layers {
   top: "decode3"
   name: "decode3"
   type: INNER_PRODUCT
-  blobs_lr: 1
-  blobs_lr: 1
-  weight_decay: 1
-  weight_decay: 0
   inner_product_param {
     num_output: 500
   }
@@ -128,10 +112,6 @@ layers {
   top: "decode2"
   name: "decode2"
   type: INNER_PRODUCT
-  blobs_lr: 1
-  blobs_lr: 1
-  weight_decay: 1
-  weight_decay: 0
   inner_product_param {
     num_output: 1000
   }