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);
}
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);
}