Add ONNX support for ResNet (#17585)

* Add ONNX support for ResNet * Add ONNX test * make fix-copies
2022-06-09 14:44:27 +02:00
parent ca2a55e9df
commit 5323094a22
5 changed files with 34 additions and 2 deletions
--- a/tests/onnx/test_onnx_v2.py
+++ b/tests/onnx/test_onnx_v2.py
@@ -182,6 +182,7 @@ PYTORCH_EXPORT_MODELS = {
    ("convbert", "YituTech/conv-bert-base"),
    ("distilbert", "distilbert-base-cased"),
    ("electra", "google/electra-base-generator"),
+    ("resnet", "microsoft/resnet-50"),
    ("roberta", "roberta-base"),
    ("roformer", "junnyu/roformer_chinese_base"),
    ("squeezebert", "squeezebert/squeezebert-uncased"),