"examples/tuners/git@developer.sourcefind.cn:OpenDAS/nni.git" did not exist on "db20820760772f44736ca4342f7ad3f9e40cca25"
Unverified Commit 31c799a0 authored by Funtowicz Morgan's avatar Funtowicz Morgan Committed by GitHub
Browse files

Tag onnx export tests as slow (#4432)

parent 8581a670
...@@ -3,7 +3,7 @@ from os import sep ...@@ -3,7 +3,7 @@ from os import sep
from os.path import dirname, exists from os.path import dirname, exists
from shutil import rmtree from shutil import rmtree
from tests.utils import require_tf, require_torch from tests.utils import require_tf, require_torch, slow
from transformers import BertConfig, BertTokenizerFast, FeatureExtractionPipeline from transformers import BertConfig, BertTokenizerFast, FeatureExtractionPipeline
from transformers.convert_graph_to_onnx import convert, ensure_valid_input, infer_shapes from transformers.convert_graph_to_onnx import convert, ensure_valid_input, infer_shapes
...@@ -22,11 +22,13 @@ class OnnxExportTestCase(unittest.TestCase): ...@@ -22,11 +22,13 @@ class OnnxExportTestCase(unittest.TestCase):
MODEL_TO_TEST = ["bert-base-cased", "gpt2", "roberta-base"] MODEL_TO_TEST = ["bert-base-cased", "gpt2", "roberta-base"]
@require_tf @require_tf
@slow
def test_export_tensorflow(self): def test_export_tensorflow(self):
for model in OnnxExportTestCase.MODEL_TO_TEST: for model in OnnxExportTestCase.MODEL_TO_TEST:
self._test_export(model, "tf", 11) self._test_export(model, "tf", 11)
@require_torch @require_torch
@slow
def test_export_pytorch(self): def test_export_pytorch(self):
for model in OnnxExportTestCase.MODEL_TO_TEST: for model in OnnxExportTestCase.MODEL_TO_TEST:
self._test_export(model, "pt", 11) self._test_export(model, "pt", 11)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment