From 99de0b655fb99e9098faeb11bc3feaf4cb46607e Mon Sep 17 00:00:00 2001 From: marcus Date: Fri, 30 Oct 2015 19:59:33 +0100 Subject: [PATCH] Use the new optimizer interface. --- src/mlpack/methods/ann/optimizer/steepest_descent.hpp | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/src/mlpack/methods/ann/optimizer/steepest_descent.hpp b/src/mlpack/methods/ann/optimizer/steepest_descent.hpp index 5639979448..44b18b744b 100644 --- a/src/mlpack/methods/ann/optimizer/steepest_descent.hpp +++ b/src/mlpack/methods/ann/optimizer/steepest_descent.hpp @@ -34,7 +34,7 @@ class SteepestDescent * @param mom The momentum coefficient. */ SteepestDescent(DecomposableFunctionType& function, - const double lr = 1, + const double lr = 0.5, const double mom = 0) : function(function), lr(lr), @@ -66,13 +66,12 @@ class SteepestDescent { if (gradient.n_elem != 0) { - DataType outputGradient; - function.Gradient(outputGradient); + DataType outputGradient = function.Gradient(); gradient += outputGradient; } else { - function.Gradient(gradient); + gradient = function.Gradient(); } }