Add SegFormer ONNX support (#18006)

* Add ONNX support * Make height and width dynamic axes Co-authored-by: Niels Rogge <nielsrogge@Nielss-MacBook-Pro.local>
2022-08-31 20:58:44 +02:00
parent 89514f0541
commit 7e7f743481
5 changed files with 38 additions and 2 deletions
--- a/docs/source/en/serialization.mdx
+++ b/docs/source/en/serialization.mdx
@@ -90,6 +90,7 @@ Ready-made configurations include the following architectures:
 - ResNet
 - RoBERTa
 - RoFormer
+- SegFormer
 - SqueezeBERT
 - T5
 - ViT