From 1065c3b7b934167943ea7824a5dae9931318c358 Mon Sep 17 00:00:00 2001 From: xaedes Date: Mon, 3 Jul 2023 18:35:11 +0200 Subject: [PATCH] tighten abs error bounds for cross_entropy_loss in test-grad0 --- tests/test-grad0.c | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test-grad0.c b/tests/test-grad0.c index fe2ca212f..0bbeff270 100644 --- a/tests/test-grad0.c +++ b/tests/test-grad0.c @@ -1386,7 +1386,7 @@ int main(int argc, const char ** argv) { struct ggml_tensor * f = ggml_cross_entropy_loss(ctx0, x[0], x[1]); - check_gradient("cross_entropy_loss", ctx0, x, f, ndims, nargs, 1e-4f, 1e-1f, INFINITY); + check_gradient("cross_entropy_loss", ctx0, x, f, ndims, nargs, 1e-4f, 1e-3f, INFINITY); } }