diff options
Diffstat (limited to 'src/caffe/layers/dropout_layer.cpp')
-rw-r--r-- | src/caffe/layers/dropout_layer.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/caffe/layers/dropout_layer.cpp b/src/caffe/layers/dropout_layer.cpp index e0068fed..e28cab33 100644 --- a/src/caffe/layers/dropout_layer.cpp +++ b/src/caffe/layers/dropout_layer.cpp @@ -20,7 +20,7 @@ void DropoutLayer<Dtype>::SetUp(const vector<Blob<Dtype>*>& bottom, DCHECK(threshold_ > 0.); DCHECK(threshold_ < 1.); scale_ = 1. / (1. - threshold_); - uint_thres_ = (unsigned int)(UINT_MAX * threshold_); + uint_thres_ = static_cast<unsigned int>(UINT_MAX * threshold_); } template <typename Dtype> @@ -37,7 +37,7 @@ Dtype DropoutLayer<Dtype>::Forward_cpu(const vector<Blob<Dtype>*>& bottom, top_data[i] = bottom_data[i] * mask[i] * scale_; } } else { - memcpy(top_data, bottom_data, bottom[0]->count() * sizeof(Dtype)); + caffe_copy(bottom[0]->count(), bottom_data, top_data); } return Dtype(0); } |