From efae2fd1f6dbf45e3b6ead75fd8cd31cd5df3d79 Mon Sep 17 00:00:00 2001 From: Adam Kropp <30504149+akropp@users.noreply.github.com> Date: Wed, 1 Nov 2023 17:26:16 -0400 Subject: [PATCH] Update src/mlpack/methods/ann/ffn_impl.hpp Co-authored-by: Ryan Curtin --- src/mlpack/methods/ann/ffn_impl.hpp | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/mlpack/methods/ann/ffn_impl.hpp b/src/mlpack/methods/ann/ffn_impl.hpp index 596213eb7d..8f22d2a9c7 100644 --- a/src/mlpack/methods/ann/ffn_impl.hpp +++ b/src/mlpack/methods/ann/ffn_impl.hpp @@ -503,9 +503,10 @@ typename MatType::elem_type FFN< MakeAlias(responsesBatch, responses.colptr(begin), responses.n_rows, batchSize); - network.Forward(predictors_batch, networkOutput); + network.Forward(predictorsBatch, networkOutput); - const typename MatType::elem_type obj = outputLayer.Forward(networkOutput, responses_batch) + network.Loss(); + const typename MatType::elem_type obj = outputLayer.Forward(networkOutput, + responsesBatch) + network.Loss(); // Now perform the backward pass. outputLayer.Backward(networkOutput, responses_batch, error);