Fixed switch and test l2hingeloss
authorSergio <sguada@gmail.com>
Sat, 21 Jun 2014 01:34:24 +0000 (18:34 -0700)
committerSergio <sguada@gmail.com>
Sat, 21 Jun 2014 01:34:24 +0000 (18:34 -0700)
Conflicts:
src/caffe/layers/loss_layer.cpp
src/caffe/test/test_hinge_loss_layer.cpp

src/caffe/test/test_hinge_loss_layer.cpp

index ec74b03..64657c8 100644 (file)
@@ -79,7 +79,7 @@ TYPED_TEST(HingeLossLayerTest, TestGradientL2CPU) {
   Caffe::set_mode(Caffe::CPU);
   HingeLossLayer<TypeParam> layer(layer_param);
   layer.SetUp(this->blob_bottom_vec_, &this->blob_top_vec_);
-  GradientChecker<TypeParam> checker(1e-2, 2e-3, 1701, 1, 0.01);
+  GradientChecker<TypeParam> 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<TypeParam> layer(layer_param);
   layer.SetUp(this->blob_bottom_vec_, &this->blob_top_vec_);
-  GradientChecker<TypeParam> checker(1e-2, 2e-3, 1701, 1, 0.01);
+  GradientChecker<TypeParam> checker(1e-2, 2e-3, 1701);
   checker.CheckGradientSingle(&layer, &(this->blob_bottom_vec_),
       &(this->blob_top_vec_), 0, -1, -1);
 }