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

Fix HubertRobustTest PT/TF equivalence test on GPU (#16943)

Browse Source 
Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar
2022-04-27 10:50:03 +02:00
committed by GitHub
parent 479fdc4925
commit 49d5bcb0f3
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
tests/hubert/test_modeling_hubert.py
View File

@@ -133,6 +133,7 @@ class HubertModelTester:
hidden_act=self.hidden_act, hidden_act=self.hidden_act,
initializer_range=self.initializer_range, initializer_range=self.initializer_range,
vocab_size=self.vocab_size, vocab_size=self.vocab_size,
do_stable_layer_norm=self.do_stable_layer_norm,
) )
def create_and_check_model(self, config, input_values, attention_mask): def create_and_check_model(self, config, input_values, attention_mask):