quantization - bugfixes

[jacinto-ai/caffe-jacinto.git] / src / caffe / layers / conv_layer.cu

diff --git a/src/caffe/layers/conv_layer.cu b/src/caffe/layers/conv_layer.cu
index 50ef113e573e32e5818bf3134b40c44fa0e14aa8..791e5d97f41a289a8d9089b01bd5e1182db00cd9 100644 (file)
--- a/src/caffe/layers/conv_layer.cu
+++ b/src/caffe/layers/conv_layer.cu
@@ -7,7 +7,7 @@ namespace caffe {
 template <typename Ftype, typename Btype>
 void ConvolutionLayer<Ftype, Btype>::Forward_gpu(const vector<Blob*>& bottom,
       const vector<Blob*>& top) {
-  //this->Quantize_gpu(bottom, top);
+  this->Quantize_gpu(bottom, top);
   const Ftype* weight = this->blobs_[0]->template gpu_data<Ftype>();
   for (int i = 0; i < bottom.size(); ++i) {
     const Ftype* bottom_data = bottom[i]->gpu_data<Ftype>();