[mlpack-git] master: Set the correct evaluation mode. (4a97187)

gitdub at big.cc.gt.atl.ga.us gitdub at big.cc.gt.atl.ga.us
Thu Jul 9 14:54:47 EDT 2015


Repository : https://github.com/mlpack/mlpack

On branch  : master
Link       : https://github.com/mlpack/mlpack/compare/baeb3885b27d3b7dd552c638c605e034b1388cad...4a97187bbba7ce8a6191b714949dd818ef0f37d2

>---------------------------------------------------------------

commit 4a97187bbba7ce8a6191b714949dd818ef0f37d2
Author: Marcus Edel <marcus.edel at fu-berlin.de>
Date:   Thu Jul 9 19:57:12 2015 +0200

    Set the correct evaluation mode.


>---------------------------------------------------------------

4a97187bbba7ce8a6191b714949dd818ef0f37d2
 src/mlpack/methods/ann/cnn.hpp | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/src/mlpack/methods/ann/cnn.hpp b/src/mlpack/methods/ann/cnn.hpp
index b5700d1..f4b854c 100644
--- a/src/mlpack/methods/ann/cnn.hpp
+++ b/src/mlpack/methods/ann/cnn.hpp
@@ -66,8 +66,13 @@ class CNN
                      ErrorType& error)
     {
       deterministic = false;
+      ResetActivations(network);
       seqNum++;
-      trainError += Evaluate(input, target, error);
+
+      std::get<0>(std::get<0>(network)).InputLayer().InputActivation() = input;
+
+      LayerForward(network);
+      trainError += OutputError(network, target, error);
     }
 
     /**
@@ -130,7 +135,7 @@ class CNN
                     const OutputType& target,
                     ErrorType& error)
     {
-      deterministic = false;
+      deterministic = true;
       ResetActivations(network);
 
       std::get<0>(std::get<0>(network)).InputLayer().InputActivation() = input;



More information about the mlpack-git mailing list