Turn atol to 1e-5 when comparing the end to end results (#17708)

Summary:
results smaller than 1e-5 don't make sense.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/17708

Differential Revision: D14348893

Pulled By: houseroad

fbshipit-source-id: 5e07c38e5b58b27b61fae63bfc3c21e2fe5629fe

diff --git a/test/onnx/verify.py b/test/onnx/verify.py
index c8521cd..b92926a 100644 (file)
--- a/test/onnx/verify.py
+++ b/test/onnx/verify.py
@@ -29,7 +29,7 @@ class Errors(object):
     >>>     ...
     """
 
-    def __init__(self, msg, rtol=1e-3, atol=1e-7):
+    def __init__(self, msg, rtol=1e-3, atol=1e-5):
         self.msg = msg
         self.errors = []
         self.context = []