From 19be78b2b610eb86aa33f3a89cb03abf58317d63 Mon Sep 17 00:00:00 2001
From: Zhengshou Lai <lai.zhengshou@outlook.com>
Date: Sun, 11 May 2025 21:50:06 +0800
Subject: [PATCH] fixes

---
 src/mlpack/methods/ann/ffn_impl.hpp | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

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<typename OutputLayerType,