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

Use larger atol in torch.allclose for some tests (#21966)

Browse Source 
Use larger atol

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar
2023-03-06 17:41:00 +01:00
committed by GitHub
parent 64d95c44ec
commit 9474abdf47
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/test_modeling_common.py
View File

@@ -2497,7 +2497,7 @@ class ModelTesterMixin:
torch.manual_seed(0) torch.manual_seed(0)
new_output = new_model(**inputs_dict_class) new_output = new_model(**inputs_dict_class)
self.assertTrue(torch.allclose(base_output[0], new_output[0])) self.assertTrue(torch.allclose(base_output[0], new_output[0], atol=1e-5))
@require_accelerate @require_accelerate
@mark.accelerate_tests @mark.accelerate_tests
@@ -2533,7 +2533,7 @@ class ModelTesterMixin:
torch.manual_seed(0) torch.manual_seed(0)
new_output = new_model(**inputs_dict_class) new_output = new_model(**inputs_dict_class)
self.assertTrue(torch.allclose(base_output[0], new_output[0])) self.assertTrue(torch.allclose(base_output[0], new_output[0], atol=1e-5))
@require_accelerate @require_accelerate
@mark.accelerate_tests @mark.accelerate_tests
@@ -2569,7 +2569,7 @@ class ModelTesterMixin:
torch.manual_seed(0) torch.manual_seed(0)
new_output = new_model(**inputs_dict_class) new_output = new_model(**inputs_dict_class)
self.assertTrue(torch.allclose(base_output[0], new_output[0])) self.assertTrue(torch.allclose(base_output[0], new_output[0], atol=1e-5))
def test_problem_types(self): def test_problem_types(self):
config, inputs_dict = self.model_tester.prepare_config_and_inputs_for_common() config, inputs_dict = self.model_tester.prepare_config_and_inputs_for_common()