projects
/
platform
/
upstream
/
caffe.git
/ commitdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
| commitdiff |
tree
raw
|
patch
| inline |
side by side
(parent:
46b79e1
)
Fixed parameter order
author
Sergio
<sguada@gmail.com>
Mon, 14 Apr 2014 03:14:43 +0000
(20:14 -0700)
committer
Jeff Donahue
<jeff.donahue@gmail.com>
Sat, 24 May 2014 22:15:11 +0000
(15:15 -0700)
src/caffe/layers/pooling_layer.cu
patch
|
blob
|
history
diff --git
a/src/caffe/layers/pooling_layer.cu
b/src/caffe/layers/pooling_layer.cu
index
77126e6
..
8be1521
100644
(file)
--- a/
src/caffe/layers/pooling_layer.cu
+++ b/
src/caffe/layers/pooling_layer.cu
@@
-319,6
+319,7
@@
void PoolingLayer<Dtype>::Backward_gpu(const vector<Blob<Dtype>*>& 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()) {