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

Fixed minor bug

Browse Source
This commit is contained in:
Praveen Ch
2018-03-16 19:57:50 +05:30
committed by manthan-r-sheth
parent af7a45a244
commit 021c1a7263
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/layer/batchnorm_impl.hpp
+3 -2
View File
@@ -127,8 +127,9 @@ void BatchNorm<InputDataType, OutputDataType>::Gradient(
for (size_t i = 0; i < normalized.n_rows; i++)
{
normalized.row(i) = (input.row(i) - arma::as_scalar(arma::mean(input.row(i), 1)));
normalized.row(i) /= (arma::as_scalar(arma::sqrt(variance.row(i) + eps)));
normalized.row(i) = input.row(i) - arma::as_scalar(arma::mean(input.row(i), 1));
normalized.row(i) /=
arma::as_scalar(arma::sqrt(arma::var(input.row(i), 1, 1) + eps));
}
gradient.submat(0, 0, gamma.n_elem - 1, 0) = arma::sum(normalized % error, 1);