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

fix

Browse Source
This commit is contained in:
Zhengshou Lai
2025-05-11 22:45:33 +08:00
parent 19be78b2b6
commit 21f23b3058
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/mlpack/methods/ann/ffn_impl.hpp
+1 -1
View File
@@ -232,7 +232,7 @@ void FFN<
}
// Perform the backward pass.
networkDelta.set_size(predictors.n_rows, batchSize);
networkDelta.set_size(predictors.n_rows, 1);
network.Backward(networkOutput, error, gradients, networkDelta);
}