diff --git a/src/mlpack/methods/ann/ffn_impl.hpp b/src/mlpack/methods/ann/ffn_impl.hpp index b7749bf9a1..10c5c215a8 100644 --- a/src/mlpack/methods/ann/ffn_impl.hpp +++ b/src/mlpack/methods/ann/ffn_impl.hpp @@ -232,7 +232,8 @@ void FFN< } // Perform the backward pass. - network.Backward(networkOutput, error, gradients); + networkDelta.set_size(predictors.n_rows, batchSize); + network.Backward(networkOutput, error, gradients, networkDelta); } template