[mlpack-git] master: Use the correct filter/kernel parameter. (236d5bc)
gitdub at big.cc.gt.atl.ga.us
gitdub at big.cc.gt.atl.ga.us
Tue Sep 1 03:58:14 EDT 2015
Repository : https://github.com/mlpack/mlpack
On branch : master
Link : https://github.com/mlpack/mlpack/compare/7b68ca3376ccf081e82146ef02710eedcd4f3aa8...236d5bcda2680cf2c5cae3ce6db7c342cad2842b
>---------------------------------------------------------------
commit 236d5bcda2680cf2c5cae3ce6db7c342cad2842b
Author: Marcus Edel <marcus.edel at fu-berlin.de>
Date: Mon Aug 31 17:29:27 2015 +0200
Use the correct filter/kernel parameter.
>---------------------------------------------------------------
236d5bcda2680cf2c5cae3ce6db7c342cad2842b
src/mlpack/methods/ann/layer/pooling_layer.hpp | 8 ++++----
1 file changed, 4 insertions(+), 4 deletions(-)
diff --git a/src/mlpack/methods/ann/layer/pooling_layer.hpp b/src/mlpack/methods/ann/layer/pooling_layer.hpp
index 6ec324c..d9bb072 100644
--- a/src/mlpack/methods/ann/layer/pooling_layer.hpp
+++ b/src/mlpack/methods/ann/layer/pooling_layer.hpp
@@ -68,8 +68,8 @@ class PoolingLayer
template<typename eT>
void Forward(const arma::Cube<eT>& input, arma::Cube<eT>& output)
{
- output = arma::zeros<arma::Cube<eT> >(input.n_rows / ksize,
- input.n_cols / ksize, input.n_slices);
+ output = arma::zeros<arma::Cube<eT> >(input.n_rows / kSize,
+ input.n_cols / kSize, input.n_slices);
for (size_t s = 0; s < input.n_slices; s++)
Pooling(input.slice(s), output.slice(s));
@@ -175,8 +175,8 @@ class PoolingLayer
void Pooling(const arma::Mat<eT>& input, arma::Mat<eT>& output)
{
- const size_t rStep = ksize;
- const size_t cStep = ksize;
+ const size_t rStep = kSize;
+ const size_t cStep = kSize;
for (size_t j = 0; j < input.n_cols; j += cStep)
{
More information about the mlpack-git
mailing list