From e08651da2c73be3a3803e86b812f545c17105b52 Mon Sep 17 00:00:00 2001 From: Sergio Date: Sun, 13 Apr 2014 20:14:43 -0700 Subject: [PATCH] Fixed parameter order --- src/caffe/layers/pooling_layer.cu | 1 + 1 file changed, 1 insertion(+) diff --git a/src/caffe/layers/pooling_layer.cu b/src/caffe/layers/pooling_layer.cu index 77126e6..8be1521 100644 --- a/src/caffe/layers/pooling_layer.cu +++ b/src/caffe/layers/pooling_layer.cu @@ -319,6 +319,7 @@ void PoolingLayer::Backward_gpu(const vector*>& top, const Dtype* top_diff = top[0]->gpu_diff(); Dtype* bottom_diff = (*bottom)[0]->mutable_gpu_diff(); int count = (*bottom)[0]->count(); + CUDA_CHECK(cudaMemset(bottom_diff, 0, sizeof(Dtype) * count)); int* mask; switch (this->layer_param_.pooling_param().pool()) { -- 2.7.4