euclidean layer update
[jacinto-ai/caffe-jacinto.git] / src / caffe / test / test_multinomial_logistic_loss_layer.cpp
index de50245d8b1fe113457cf2a71e11950c13971c46..5595c84fea3fc9e05b3ded302f7c48f4f09db858 100644 (file)
@@ -52,6 +52,7 @@ TYPED_TEST(MultinomialLogisticLossLayerTest, TestGradientCPU) {
   LayerParameter layer_param;
   Caffe::set_mode(Caffe::CPU);
   MultinomialLogisticLossLayer<TypeParam> layer(layer_param);
+  layer.SetUp(this->blob_bottom_vec_, &this->blob_top_vec_);
   GradientChecker<TypeParam> checker(1e-2, 1e-2, 1701, 0, 0.05);
   checker.CheckGradientSingle(layer, this->blob_bottom_vec_,
       this->blob_top_vec_, 0, -1, -1);