fixes

2025-05-11 21:50:06 +08:00
parent 665479fde2
commit 19be78b2b6
1 changed files with 2 additions and 1 deletions
@@ -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,