[go: nahoru, domu]

Skip to content

Commit

Permalink
Minor style update
Browse files Browse the repository at this point in the history
  • Loading branch information
Manthan-R-Sheth committed Mar 16, 2018
1 parent 37603c1 commit f46b131
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion src/mlpack/methods/ann/layer/batch_norm_impl.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -83,7 +83,7 @@ void BatchNorm<InputDataType, OutputDataType>::Forward(
template<typename InputDataType, typename OutputDataType>
template<typename eT>
void BatchNorm<InputDataType, OutputDataType>::Backward(
const arma::Mat<eT>&& input, arma::Mat<eT>&& gy, arma::Mat<eT>&& g)
const arma::Mat<eT>&& input, arma::Mat<eT>&& gy, arma::Mat<eT>&& g)
{
g.reshape(input.n_rows, input.n_cols);

Expand Down
2 changes: 1 addition & 1 deletion src/mlpack/tests/ann_layer_test.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -1353,7 +1353,7 @@ BOOST_AUTO_TEST_CASE(BatchNormTest)
model.Backward(std::move(input), std::move(gy), std::move(output));
result << -0.0780 << 0.1376 << -0.0596 << arma::endr
<< 0.0602 << -0.1317 << 0.0715 << arma::endr
<< 0.0835 << -0.1493 << 0.0658 << arma::endr;
<< 0.0835 << -0.1493 << 0.0658 << arma::endr;

CheckMatrices(output, result, 1e-1);
result.clear();
Expand Down

0 comments on commit f46b131

Please sign in to comment.