Merge pull request BVLC#3 from m1lhaus/master

Fixed test exec error: lrn_ristretto_layer.cpp:16] LRN layer only supports minifloat
cepiross · Feb 25, 2018 · 131b5b6 · 131b5b6
2 parents e2eaff7 + 2684ffe
commit 131b5b6
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/caffe/test/test_layer_factory.cpp b/src/caffe/test/test_layer_factory.cpp
@@ -42,6 +42,9 @@ TYPED_TEST(LayerFactoryTest, TestCreateLayer) {
       continue;
 #endif  // USE_LEVELDB
     }
+    if (iter->first == "LRNRistretto") {
+      layer_param.mutable_quantization_param()->set_precision(caffe::QuantizationParameter_Precision_MINIFLOAT);
+    }
     layer_param.set_type(iter->first);
     layer = LayerRegistry<Dtype>::CreateLayer(layer_param);
     EXPECT_EQ(iter->first, layer->type());