diff options
| author | Jeff Donahue <jeff.donahue@gmail.com> | 2014-04-15 15:38:06 -0700 |
|---|---|---|
| committer | Jeff Donahue <jeff.donahue@gmail.com> | 2014-04-15 15:38:06 -0700 |
| commit | 24f7ec0c34dbe62c0c207d728dbb29c09db133ec (patch) | |
| tree | afa392ae306e1073a37b47604a57986af7aa8d6c | |
| parent | 7b5a7e572db10297e33fd108d61b3aaa48a88e0f (diff) | |
| download | caffeonacl-24f7ec0c34dbe62c0c207d728dbb29c09db133ec.tar.gz caffeonacl-24f7ec0c34dbe62c0c207d728dbb29c09db133ec.tar.bz2 caffeonacl-24f7ec0c34dbe62c0c207d728dbb29c09db133ec.zip | |
mnist autoencoder test proto bugfix: add sigmoid layer before loss
| -rw-r--r-- | examples/mnist/mnist_autoencoder_test.prototxt | 10 |
1 files changed, 6 insertions, 4 deletions
diff --git a/examples/mnist/mnist_autoencoder_test.prototxt b/examples/mnist/mnist_autoencoder_test.prototxt index 440ccd3c..c106b47f 100644 --- a/examples/mnist/mnist_autoencoder_test.prototxt +++ b/examples/mnist/mnist_autoencoder_test.prototxt @@ -147,16 +147,18 @@ layers { top: "decode1" name: "decode1" type: INNER_PRODUCT - blobs_lr: 1 - blobs_lr: 1 - weight_decay: 1 - weight_decay: 0 inner_product_param { num_output: 784 } } layers { bottom: "decode1" + top: "decode1neuron" + name: "decode1neuron" + type: SIGMOID +} +layers { + bottom: "decode1neuron" bottom: "flatdata" name: "loss" type: EUCLIDEAN_LOSS |