From 8b9ae45549150b9d25dc0576eaf08562cd52158c Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Mon, 7 Mar 2022 22:14:33 +0100 Subject: [PATCH] Set scale_embedding to False in some TF tests (#15952) * set scale_embedding to False to avoid large (> 1e-5) output differences between PT/TF Co-authored-by: ydshieh --- tests/speech_to_text/test_modeling_tf_speech_to_text.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/tests/speech_to_text/test_modeling_tf_speech_to_text.py b/tests/speech_to_text/test_modeling_tf_speech_to_text.py index b9036ae3a1..0e14ab9d26 100644 --- a/tests/speech_to_text/test_modeling_tf_speech_to_text.py +++ b/tests/speech_to_text/test_modeling_tf_speech_to_text.py @@ -90,6 +90,7 @@ class TFSpeech2TextModelTester: eos_token_id=2, pad_token_id=1, bos_token_id=0, + scale_embedding=False, ): self.parent = parent self.batch_size = batch_size @@ -115,6 +116,7 @@ class TFSpeech2TextModelTester: self.eos_token_id = eos_token_id self.pad_token_id = pad_token_id self.bos_token_id = bos_token_id + self.scale_embedding = scale_embedding def prepare_config_and_inputs(self): input_features = floats_tensor( @@ -155,6 +157,7 @@ class TFSpeech2TextModelTester: eos_token_id=self.eos_token_id, bos_token_id=self.bos_token_id, pad_token_id=self.pad_token_id, + scale_embedding=self.scale_embedding, ) def prepare_config_and_inputs_for_common(self):