[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