add graph test (openvinotoolkit#734)

vshampor · May 26, 2021 · d7a3bec · d7a3bec
1 parent ad15cd0
commit d7a3bec
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 3 deletions.
diff --git a/beta/tests/tensorflow/data/reference_graphs/pruning/filter_pruning/mobilenet_v3_large.pb b/beta/tests/tensorflow/data/reference_graphs/pruning/filter_pruning/mobilenet_v3_large.pb
diff --git a/beta/tests/tensorflow/data/reference_graphs/pruning/filter_pruning/mobilenet_v3_small.pb b/beta/tests/tensorflow/data/reference_graphs/pruning/filter_pruning/mobilenet_v3_small.pb
diff --git a/beta/tests/tensorflow/data/reference_graphs/pruning/filter_pruning/yolo_v4.pb b/beta/tests/tensorflow/data/reference_graphs/pruning/filter_pruning/yolo_v4.pb
diff --git a/beta/tests/tensorflow/test_compressed_graph.py b/beta/tests/tensorflow/test_compressed_graph.py
@@ -179,9 +179,6 @@ def __init__(self, ref_graph_filename: str, model_builder, input_sample_sizes,
         'nasnet_mobile': pytest.mark.skip(reason='gitlab issue #18'),
         'xception': pytest.mark.skip(reason='gitlab issue #28'),
         'mask_rcnn': pytest.mark.skip(reason='ticket #50605'),
-        'mobilenet_v3_small': pytest.mark.skip(reason='ticket #50607'),
-        'mobilenet_v3_large': pytest.mark.skip(reason='ticket #50607'),
-        'yolo_v4': pytest.mark.skip(reason='ticket #50608'),
         'resnet50_v2': pytest.mark.skip(resason='Several masks on one weight'),
         'mobilenet_v2_slim': pytest.mark.skip(reason='ticket #46349')
     },