apaam/mlpack
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update src/mlpack/methods/ann/ffn_impl.hpp

Browse Source 
Co-authored-by: Ryan Curtin <ryan@ratml.org>
This commit is contained in:
Adam Kropp
2023-11-01 17:26:16 -04:00
committed by GitHub
co-authored by Ryan Curtin
parent 91a62a56ab
commit efae2fd1f6
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/mlpack/methods/ann/ffn_impl.hpp
+3 -2
View File
@@ -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);