Make height and width non-dynamic axes

huggingface · Jul 20, 2022 · 17e6fd0 · 17e6fd0
1 parent 2c68ac1
commit 17e6fd0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/transformers/onnx/config.py b/src/transformers/onnx/config.py
@@ -101,7 +101,7 @@ class OnnxConfig(ABC):
         ),
         "semantic-segmentation": OrderedDict(
             {
-                "logits": {0: "batch", 1: "sequence", 2: "height", 3: "width"},
+                "logits": {0: "batch", 1: "sequence"},
             }
         ),
         "seq2seq-lm": OrderedDict({"logits": {0: "batch", 1: "decoder_sequence"}}),