SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Set scale_embedding to False in some TF tests (#15952)

Browse Source 
* set scale_embedding to False to avoid large (> 1e-5) output differences between PT/TF

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar
2022-03-07 22:14:33 +01:00
committed by GitHub
parent 38cc35069c
commit 8b9ae45549
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/speech_to_text/test_modeling_tf_speech_to_text.py
View File

@@ -90,6 +90,7 @@ class TFSpeech2TextModelTester:
eos_token_id=2, eos_token_id=2,
pad_token_id=1, pad_token_id=1,
bos_token_id=0, bos_token_id=0,
scale_embedding=False,
): ):
self.parent = parent self.parent = parent
self.batch_size = batch_size self.batch_size = batch_size
@@ -115,6 +116,7 @@ class TFSpeech2TextModelTester:
self.eos_token_id = eos_token_id self.eos_token_id = eos_token_id
self.pad_token_id = pad_token_id self.pad_token_id = pad_token_id
self.bos_token_id = bos_token_id self.bos_token_id = bos_token_id
self.scale_embedding = scale_embedding
def prepare_config_and_inputs(self): def prepare_config_and_inputs(self):
input_features = floats_tensor( input_features = floats_tensor(
@@ -155,6 +157,7 @@ class TFSpeech2TextModelTester:
eos_token_id=self.eos_token_id, eos_token_id=self.eos_token_id,
bos_token_id=self.bos_token_id, bos_token_id=self.bos_token_id,
pad_token_id=self.pad_token_id, pad_token_id=self.pad_token_id,
scale_embedding=self.scale_embedding,
) )
def prepare_config_and_inputs_for_common(self): def prepare_config_and_inputs_for_common(self):