Skip to content

Commit 4c8661d

Browse files
committed
change atol from 0 to 1e-6
1 parent e795995 commit 4c8661d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/run_pretrained_models.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -84,7 +84,7 @@ class Test(object):
8484
target = []
8585

8686
def __init__(self, url, local, make_input, input_names, output_names,
87-
disabled=False, more_inputs=None, rtol=0.01, atol=0.,
87+
disabled=False, more_inputs=None, rtol=0.01, atol=1e-6,
8888
check_only_shape=False, model_type="frozen", force_input_shape=False,
8989
skip_tensorflow=False):
9090
self.url = url

0 commit comments

Comments
 (0)