Fix var names in Dropout.cu
authorSergio <sguada@gmail.com>
Sat, 14 Jun 2014 01:54:28 +0000 (18:54 -0700)
committerSergio <sguada@gmail.com>
Sat, 28 Jun 2014 01:54:21 +0000 (18:54 -0700)
src/caffe/layers/dropout_layer.cu

index 0040d26..2e0b0d3 100644 (file)
@@ -70,7 +70,7 @@ void DropoutLayer<Dtype>::Backward_gpu(const vector<Blob<Dtype>*>& top,
           count, top_diff, mask, uint_thres_, scale_, bottom_diff);
       CUDA_POST_KERNEL_CHECK;
     } else {
-      caffe_gpu_copy(top[0]->count(), bottom_data, top_data);
+      caffe_gpu_copy(top[0]->count(), top_diff, bottom_diff);
     }
   }
 }