Higher tolerance for past testing in TF T5 (#3844)

1d4a35b3 · Patrick von Platen · GitHub · d13eca11 · 1d4a35b3
Unverified Commit 1d4a35b3 authored Apr 17, 2020 by Patrick von Platen Committed by GitHub Apr 17, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

tests/test_modeling_tf_t5.py tests/test_modeling_tf_t5.py +2 -2

No files found.
--- a/tests/test_modeling_tf_t5.py
+++ b/tests/test_modeling_tf_t5.py
@@ -177,7 +177,7 @@ class TFT5ModelTest(TFModelTesterMixin, unittest.TestCase):
            output_from_past_slice = output_from_past[:, 0, random_slice_idx]

            # test that outputs are equal for slice
-            tf.debugging.assert_near(output_from_past_slice, output_from_no_past_slice, rtol=1e-6)
+            tf.debugging.assert_near(output_from_past_slice, output_from_no_past_slice, rtol=1e-3)

        def create_and_check_t5_decoder_model_attention_mask_past(
            self, config, input_ids, decoder_input_ids, attention_mask
@@ -221,7 +221,7 @@ class TFT5ModelTest(TFModelTesterMixin, unittest.TestCase):
            output_from_past_slice = output_from_past[:, 0, random_slice_idx]

            # test that outputs are equal for slice
-            tf.debugging.assert_near(output_from_past_slice, output_from_no_past_slice, rtol=1e-6)
+            tf.debugging.assert_near(output_from_past_slice, output_from_no_past_slice, rtol=1e-3)

        def prepare_config_and_inputs_for_common(self):
            config_and_inputs = self.prepare_config_and_inputs()