From b3201eb5c322f4d88860a190addc05be9b263baf Mon Sep 17 00:00:00 2001
From: Niels Rogge <nielsrogge@Nielss-MacBook-Pro.local>
Date: Sat, 2 Jul 2022 11:03:59 +0200
Subject: [PATCH] Add semantic-segmentation to config

---
 src/transformers/onnx/config.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/transformers/onnx/config.py b/src/transformers/onnx/config.py
index 6097ebf49aa27a..874d53a6757756 100644
--- a/src/transformers/onnx/config.py
+++ b/src/transformers/onnx/config.py
@@ -99,6 +99,11 @@ class OnnxConfig(ABC):
                 "end_logits": {0: "batch", 1: "sequence"},
             }
         ),
+        "semantic-segmentation": OrderedDict(
+            {
+                "logits": {0: "batch", 1: "sequence", 2: "height", 3: "width"},
+            }
+        ),
         "seq2seq-lm": OrderedDict({"logits": {0: "batch", 1: "decoder_sequence"}}),
         "sequence-classification": OrderedDict({"logits": {0: "batch"}}),
         "token-classification": OrderedDict({"logits": {0: "batch", 1: "sequence"}}),