Merge pull request #154 from zhongwen/pooling_size

fix pooling output size
dmlc · May 15, 2015 · 47660af · 47660af
2 parents 0b33b86 + 9a81b41
commit 47660af
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/layer/pooling_layer-inl.hpp b/src/layer/pooling_layer-inl.hpp
@@ -119,8 +119,8 @@ class PoolingLayer : public ILayer<xpu> {
 
     mshadow::Shape<4> oshape = mshadow::
         Shape4(ishape[0], ishape[1],
-               std::min(ishape[2] + 2 * param_.pad_y - ksize_y + kstride-1, ishape[2] + 2 * param_.pad_y - 1) / kstride + 1,
-               std::min(ishape[3] + 2 * param_.pad_x - ksize_x + kstride-1, ishape[3] + 2 * param_.pad_x- 1) / kstride + 1);
+               (ishape[2] + 2 * param_.pad_y - ksize_y) / kstride + 1,
+               (ishape[3] + 2 * param_.pad_x - ksize_x) / kstride + 1);
     nodes_out[0]->data.shape_ = oshape;
     // use 2 temp state to store pooled result (state 2 for cudnn)
     p_cstate->states.resize(2);