diff --git a/examples/resnet-18/resnet-18.py b/examples/resnet-18/resnet-18.py index d5148c6ca..203400130 100644 --- a/examples/resnet-18/resnet-18.py +++ b/examples/resnet-18/resnet-18.py @@ -65,3 +65,6 @@ def compare_labels(id0, id1): y0 = shark_infer(x) y1 = forward(x) print_labels(y0) +print( + f"Found {compare_labels(y0,y1)[0].size()[0]} discrepancies between turbine and standard result" +) diff --git a/python/turbine_models/tests/resnet_test.py b/python/turbine_models/tests/resnet_test.py index a6aea93a3..5d514e6fe 100644 --- a/python/turbine_models/tests/resnet_test.py +++ b/python/turbine_models/tests/resnet_test.py @@ -3,6 +3,7 @@ from turbine_models.custom_models import resnet_18 import unittest import os +import pytest arguments = { "run_vmfb": True, @@ -17,6 +18,9 @@ class Resnet18Test(unittest.TestCase): + @pytest.mark.xfail( + reason="caused by lack of support for DenseResourceElementsAttr iteration over a generic FloatAttr" + ) def testExportResnet18Model(self): with self.assertRaises(SystemExit) as cm: resnet_18.export_resnet_18_model(