diff --git a/src/mlpack/methods/ann/ffn_impl.hpp b/src/mlpack/methods/ann/ffn_impl.hpp index 557c4a3fdc..2deda4090d 100644 --- a/src/mlpack/methods/ann/ffn_impl.hpp +++ b/src/mlpack/methods/ann/ffn_impl.hpp @@ -252,9 +252,11 @@ void FFN< } else if (inputDimensions.size() > 0) { - const size_t inputDims = std::accumulate(inputDimensions.begin(), - inputDimensions.end(), 0); - CheckNetwork("FFN::Reset()", inputDims, true, false); + size_t inputDim = inputDimensions[0]; + for (size_t i=1; i