[mlpack-git] master: Use Gradient() to get the gradient. (ea3ebab)
gitdub at big.cc.gt.atl.ga.us
gitdub at big.cc.gt.atl.ga.us
Fri Feb 27 15:51:44 EST 2015
Repository : https://github.com/mlpack/mlpack
On branch : master
Link : https://github.com/mlpack/mlpack/compare/594fd9f61d1280152c758559b4fc60bf0c827cca...45f682337b1daa4c82797f950e16a605fe4971bd
>---------------------------------------------------------------
commit ea3ebabbc2d976f53f04944864a72efe61ec692d
Author: Marcus Edel <marcus.edel at fu-berlin.de>
Date: Wed Jan 21 14:39:05 2015 +0100
Use Gradient() to get the gradient.
>---------------------------------------------------------------
ea3ebabbc2d976f53f04944864a72efe61ec692d
src/mlpack/methods/ann/ffnn.hpp | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/src/mlpack/methods/ann/ffnn.hpp b/src/mlpack/methods/ann/ffnn.hpp
index 4b5e951..876f924 100644
--- a/src/mlpack/methods/ann/ffnn.hpp
+++ b/src/mlpack/methods/ann/ffnn.hpp
@@ -353,8 +353,9 @@ class FFNN
typename std::enable_if<I < sizeof...(Tp), void>::type
Gradients(std::tuple<Tp...>& t)
{
- gradients[gradientNum++] += std::get<I>(t).OutputLayer().Delta() *
- std::get<I>(t).InputLayer().InputActivation().t();
+ MatType gradient;
+ std::get<I>(t).Gradient(gradient);
+ gradients[gradientNum++] += gradient;
Gradients<I + 1, Tp...>(t);
}
More information about the mlpack-git
mailing list