From 5ac06bad619345a1a7d760c7d69666b31e796372 Mon Sep 17 00:00:00 2001 From: Sherlock Date: Mon, 21 Jun 2021 07:41:54 -0700 Subject: [PATCH] Relax test tolerance to make CI more reliable (#8100) --- .../orttraining/test/python/orttraining_test_ortmodule_api.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/orttraining/orttraining/test/python/orttraining_test_ortmodule_api.py b/orttraining/orttraining/test/python/orttraining_test_ortmodule_api.py index 6037267ae9..48b971ea16 100644 --- a/orttraining/orttraining/test/python/orttraining_test_ortmodule_api.py +++ b/orttraining/orttraining/test/python/orttraining_test_ortmodule_api.py @@ -890,7 +890,7 @@ def test_multiple_ortmodules_common_backbone_training(): ort_prediction = run_step(ort_model0, ort_model2, x1) _test_helpers.assert_values_are_close(ort_prediction, pt_prediction) - _test_helpers.assert_gradients_match_and_reset_gradient(ort_model0, pt_model0, reset_gradient=True, atol=1.5e-6) + _test_helpers.assert_gradients_match_and_reset_gradient(ort_model0, pt_model0, reset_gradient=True, atol=1e-5) _test_helpers.assert_gradients_match_and_reset_gradient(ort_model2, pt_model2) def test_multiple_chained_ortmodules_training():