Skip to content

GitLab

Commit 0009e79b authored by Frederick Liu's avatar Frederick Liu Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 417646384
parent 964412b9
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/nlp/modeling/layers/reuse_transformer_test.py
View file @ 0009e79b
...@@ -121,7 +121,7 @@ class ReuseTransformerLayerTest(tf.test.TestCase, parameterized.TestCase): ...@@ -121,7 +121,7 @@ class ReuseTransformerLayerTest(tf.test.TestCase, parameterized.TestCase):
new_layer.set_weights(test_layer.get_weights()) new_layer.set_weights(test_layer.get_weights())
new_output_tensor, _ = new_layer([input_data, mask_data]) new_output_tensor, _ = new_layer([input_data, mask_data])
self.assertAllClose( 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): def test_layer_output_range_with_relative_pe(self, transformer_cls):
test_layer = transformer_cls( test_layer = transformer_cls(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment