TYPED_TEST(ArgMaxLayerTest, TestSetupMaxVal) {
LayerParameter layer_param;
ArgMaxParameter* argmax_param = layer_param.mutable_argmax_param();
- argmax_param->set_out_max_val(true)
+ argmax_param->set_out_max_val(true);
ThresholdLayer<TypeParam> layer(layer_param);
layer.SetUp(this->blob_bottom_vec_, &(this->blob_top_vec_));
EXPECT_EQ(this->blob_top_->num(), this->bottom_top_->num());
LayerParameter layer_param;
Caffe::set_mode(Caffe::CPU);
ArgMaxParameter* argmax_param = layer_param.mutable_argmax_param();
- argmax_param->set_out_max_val(true)
+ argmax_param->set_out_max_val(true);
ArgMaxLayer<TypeParam> layer(layer_param);
layer.SetUp(this->blob_bottom_vec_, &(this->blob_top_vec_));
layer.Forward(this->blob_bottom_vec_, &(this->blob_top_vec_));