diff --git a/src/mlpack/methods/ann/layer/batch_norm.hpp b/src/mlpack/methods/ann/layer/batch_norm.hpp index 7b20a9c27e..05b1f1a1af 100644 --- a/src/mlpack/methods/ann/layer/batch_norm.hpp +++ b/src/mlpack/methods/ann/layer/batch_norm.hpp @@ -180,10 +180,6 @@ class BatchNorm //! Locally-stored value for momentum. double momentum; - //! Locally-stored value for average factor which used to update running - //! mean and variance. - double averageFactor; - //! Variable to keep track of whether we are in loading or saving mode. bool loading; @@ -211,6 +207,10 @@ class BatchNorm //! Locally-stored running mean/variance counter. size_t count; + //! Locally-stored value for average factor which used to update running + //! mean and variance. + double averageFactor; + //! Locally-stored mean object. OutputDataType runningMean;