diff --git a/tensorflow/python/keras/distribute/keras_correctness_test_base.py b/tensorflow/python/keras/distribute/keras_correctness_test_base.py index 9d66754a845..4a855f60777 100644 --- a/tensorflow/python/keras/distribute/keras_correctness_test_base.py +++ b/tensorflow/python/keras/distribute/keras_correctness_test_base.py @@ -308,7 +308,7 @@ def compare_results(results_with_ds, default_tolerance = 1e-3 relaxed_tolerance = 1e-3 else: - default_tolerance = 4e-5 + default_tolerance = 1e-5 relaxed_tolerance = 1e-4 def _get_compare_result_tolerance(key):