rtol -> atol

PaddlePaddle · Feb 1, 2023 · a636444 · a636444
1 parent 1033445
commit a636444
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/tests/transformers/bert/test_modeling.py b/tests/transformers/bert/test_modeling.py
@@ -737,7 +737,7 @@ def test_bert_classes_from_local_dir(self, class_name, pytorch_class_name: str |
                 np.allclose(
                     paddle_logit.detach().cpu().reshape([-1])[:9].numpy(),
                     torch_logit.detach().cpu().reshape([-1])[:9].numpy(),
-                    rtol=1e-4,
+                    atol=1e-3,
                 )
             )
 

diff --git a/tests/transformers/gpt/test_modeling.py b/tests/transformers/gpt/test_modeling.py
@@ -627,7 +627,7 @@ def test_gpt_classes_from_local_dir(self, paddle_class_name, pytorch_class_name:
                 np.allclose(
                     paddle_logit.detach().cpu().numpy().reshape([-1])[:9],
                     torch_logit.detach().cpu().numpy().reshape([-1])[:9],
-                    rtol=1e-3,
+                    atol=1e-3,
                 )
             )
 

diff --git a/tests/transformers/roberta/test_modeling.py b/tests/transformers/roberta/test_modeling.py
@@ -500,7 +500,7 @@ def test_roberta_classes_from_local_dir(self, class_name):
                 np.allclose(
                     paddle_logit.detach().cpu().reshape([-1])[:9].numpy(),
                     torch_logit.detach().cpu().reshape([-1])[:9].numpy(),
-                    rtol=1e-4,
+                    atol=1e-3,
                 )
             )
 

diff --git a/tests/transformers/t5/test_modeling.py b/tests/transformers/t5/test_modeling.py
@@ -822,7 +822,7 @@ def test_t5_for_conditional_generation(self):
                 np.allclose(
                     paddle_logit.detach().cpu().reshape([-1])[:9].numpy(),
                     torch_logit.detach().cpu().reshape([-1])[:9].numpy(),
-                    rtol=1e-4,
+                    atol=1e-3,
                 )
             )