diff --git a/libs/ml/tests/neural_net/gradient.cpp b/libs/ml/tests/neural_net/gradient.cpp index d86a6497..6764409b 100644 --- a/libs/ml/tests/neural_net/gradient.cpp +++ b/libs/ml/tests/neural_net/gradient.cpp @@ -75,7 +75,7 @@ test_case(ml_neural__net_gradient) nn.weights()[i] = old; double numeric_gradient = (v1 - v0) / 2.0 / eps; - expect_close(numeric_gradient, learner.gradient()[i], 1e-4); + expect_close(numeric_gradient, learner.gradient()[i], 1e-6); } } } @@ -122,7 +122,7 @@ test_case(ml_neural__net_arg__gradient) input[i] = old; double numeric_gradient = (v1 - v0) / 2.0 / eps; - expect_close(numeric_gradient, learner.arg_gradient()[i], 1e-4); + expect_close(numeric_gradient, learner.arg_gradient()[i], 1e-6); } } }