From 7be676cfb743bd2dddb2b070da5bf5c52a9cd8d5 Mon Sep 17 00:00:00 2001 From: Sergio Date: Fri, 20 Jun 2014 18:34:24 -0700 Subject: [PATCH] Fixed switch and test l2hingeloss Conflicts: src/caffe/layers/loss_layer.cpp src/caffe/test/test_hinge_loss_layer.cpp --- src/caffe/test/test_hinge_loss_layer.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/caffe/test/test_hinge_loss_layer.cpp b/src/caffe/test/test_hinge_loss_layer.cpp index ec74b03..64657c8 100644 --- a/src/caffe/test/test_hinge_loss_layer.cpp +++ b/src/caffe/test/test_hinge_loss_layer.cpp @@ -79,7 +79,7 @@ TYPED_TEST(HingeLossLayerTest, TestGradientL2CPU) { Caffe::set_mode(Caffe::CPU); HingeLossLayer layer(layer_param); layer.SetUp(this->blob_bottom_vec_, &this->blob_top_vec_); - GradientChecker checker(1e-2, 2e-3, 1701, 1, 0.01); + GradientChecker checker(1e-2, 2e-3, 1701); checker.CheckGradientSingle(&layer, &(this->blob_bottom_vec_), &(this->blob_top_vec_), 0, -1, -1); } @@ -93,7 +93,7 @@ TYPED_TEST(HingeLossLayerTest, TestGradientL2GPU) { Caffe::set_mode(Caffe::GPU); HingeLossLayer layer(layer_param); layer.SetUp(this->blob_bottom_vec_, &this->blob_top_vec_); - GradientChecker checker(1e-2, 2e-3, 1701, 1, 0.01); + GradientChecker checker(1e-2, 2e-3, 1701); checker.CheckGradientSingle(&layer, &(this->blob_bottom_vec_), &(this->blob_top_vec_), 0, -1, -1); } -- 2.7.4