Skip to content

Commit c6a24fd

Browse files
authored
turn rtol to 0.002 on densenet121, since AMD and Nvidia GPU's precion difference (#1733)
1 parent 5b7ac72 commit c6a24fd

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

onnx/backend/test/case/model/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,7 @@ def collect_testcases(): # type: () -> List[TestCase]
4747

4848
model_tests = [
4949
('test_bvlc_alexnet', 'bvlc_alexnet', 1e-3, 1e-7),
50-
('test_densenet121', 'densenet121', 1e-3, 1e-7),
50+
('test_densenet121', 'densenet121', 2e-3, 1e-7),
5151
('test_inception_v1', 'inception_v1', 1e-3, 1e-7),
5252
('test_inception_v2', 'inception_v2', 1e-3, 1e-7),
5353
('test_resnet50', 'resnet50', 1e-3, 1e-7),

0 commit comments

Comments
 (0)