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

Update src/mlpack/methods/ann/layer/multi_layer_impl.hpp

Browse Source 
Co-authored-by: Ryan Curtin <ryan@ratml.org>
This commit is contained in:
Adam Kropp
2023-11-01 18:03:54 -04:00
committed by GitHub
co-authored by Ryan Curtin
parent b35fbc05c5
commit 83f89f7b1b
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/mlpack/methods/ann/layer/multi_layer_impl.hpp
+2 -1
View File
@@ -187,7 +187,8 @@ void MultiLayer<MatType>::Backward(
network.back()->Backward(layerOutputs[network.size() - 2], output, gy, layerDeltas.back());
for (size_t i = network.size() - 2; i > 0; --i)
network[i]->Backward(layerOutputs[i-1], layerOutputs[i], layerDeltas[i + 1], layerDeltas[i]);
network[i]->Backward(layerOutputs[i - 1], layerOutputs[i],
layerDeltas[i + 1], layerDeltas[i]);
network[0]->Backward(input, layerOutputs[0], layerDeltas[1], g);
}
else if (network.size() == 1)