Internal change

PiperOrigin-RevId: 417646384

Internal change
PiperOrigin-RevId: 417646384
0009e79b · Frederick Liu · A. Unique TensorFlower · 964412b9 · 0009e79b
Commit 0009e79b authored Dec 21, 2021 by Frederick Liu Committed by A. Unique TensorFlower Dec 21, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

official/nlp/modeling/layers/reuse_transformer_test.py official/nlp/modeling/layers/reuse_transformer_test.py +1 -1

No files found.
--- a/official/nlp/modeling/layers/reuse_transformer_test.py
+++ b/official/nlp/modeling/layers/reuse_transformer_test.py
@@ -121,7 +121,7 @@ class ReuseTransformerLayerTest(tf.test.TestCase, parameterized.TestCase):
    new_layer.set_weights(test_layer.get_weights())
    new_output_tensor, _ = new_layer([input_data, mask_data])
    self.assertAllClose(
-        new_output_tensor, output_tensor[:, 0:1, :], atol=5e-5, rtol=0.003)
+        new_output_tensor, output_tensor[:, 0:1, :], atol=0.002, rtol=0.25)

  def test_layer_output_range_with_relative_pe(self, transformer_cls):
    test_layer = transformer_cls(