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

Move threshold up for flaky test with Electra (#6622)

Browse Source 
* Move threshold up for flaky test with Electra

* Update above as well
This commit is contained in:
Sylvain Gugger
2020-08-20 13:59:40 -04:00
committed by GitHub
parent 953958372a
commit e8af90c052
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/test_modeling_tf_common.py
View File

@@ -306,13 +306,13 @@ class TFModelTesterMixin:
max_diff = np.amax(np.abs(tf_hidden_states - pt_hidden_states)) max_diff = np.amax(np.abs(tf_hidden_states - pt_hidden_states))
# Debug info (remove when fixed) # Debug info (remove when fixed)
if max_diff >= 2e-2: if max_diff >= 4e-2:
print("===") print("===")
print(model_class) print(model_class)
print(config) print(config)
print(inputs_dict) print(inputs_dict)
print(pt_inputs_dict) print(pt_inputs_dict)
self.assertLessEqual(max_diff, 2e-2) self.assertLessEqual(max_diff, 4e-2)
# Check we can load pt model in tf and vice-versa with checkpoint => model functions # Check we can load pt model in tf and vice-versa with checkpoint => model functions
with tempfile.TemporaryDirectory() as tmpdirname: with tempfile.TemporaryDirectory() as tmpdirname: