From 70efb6886fdac7a2d79f6155df2569df4c2cfb99 Mon Sep 17 00:00:00 2001 From: Sourabh Varshney Date: Sat, 16 Dec 2017 18:51:12 +0530 Subject: [PATCH] Some more style changes --- .../core/optimizers/adam/nadam_update.hpp | 17 +++++++++-------- 1 file changed, 9 insertions(+), 8 deletions(-) diff --git a/src/mlpack/core/optimizers/adam/nadam_update.hpp b/src/mlpack/core/optimizers/adam/nadam_update.hpp index eeb3757d42..752ba87937 100644 --- a/src/mlpack/core/optimizers/adam/nadam_update.hpp +++ b/src/mlpack/core/optimizers/adam/nadam_update.hpp @@ -43,17 +43,18 @@ class NadamUpdate * parameter. * @param beta1 The smoothing parameter. * @param beta2 The second moment coefficient + * @param scheduleDecay The decay parameter for decay coefficients */ NadamUpdate(const double epsilon = 1e-8, const double beta1 = 0.9, const double beta2 = 0.99, const double scheduleDecay = 4e-3) - :epsilon(epsilon), - beta1(beta1), - beta2(beta2), - scheduleDecay(scheduleDecay), - iteration(0), - cumBeta1(1) + :epsilon(epsilon), + beta1(beta1), + beta2(beta2), + scheduleDecay(scheduleDecay), + iteration(0), + cumBeta1(1) { // Nothing to do. } @@ -132,9 +133,9 @@ class NadamUpdate //! Modify the second moment coefficient. double& Beta2() { return beta2; } - //! Get the dacay parameter for decay coefficients + //! Get the decay parameter for decay coefficients double ScheduleDecay() const { return scheduleDecay; } - //! Modify the dacay parameter for decay coefficients + //! Modify the decay parameter for decay coefficients double& ScheduleDecay() { return scheduleDecay; } private: