Skip to content
This repository has been archived by the owner on Aug 5, 2022. It is now read-only.

Commit

Permalink
Fix the threshod of failed unit test.
Browse files Browse the repository at this point in the history
  • Loading branch information
Yu, Chong committed Apr 26, 2017
1 parent 102bf13 commit 20299f2
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/caffe/test/test_mkldnn_batch_norm_layer.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -170,7 +170,7 @@ namespace caffe {
LayerParameter layer_param;

MKLDNNBatchNormLayer<Dtype> layer(layer_param);
GradientChecker<Dtype> checker(1e-2, 1e-4);
GradientChecker<Dtype> checker(1e-2, 1e-3);
checker.CheckGradientExhaustive(&layer, this->blob_bottom_vec_,
this->blob_top_vec_);
}
Expand Down

0 comments on commit 20299f2

Please sign in to comment.