Update transformers architecture class (#1)

- Update transformers architecture class (5ee01c08e115cbd6737b9925745d900a37755fa9)
2022-07-19 10:40:37 +00:00 · 2022-07-19 10:40:37 +00:00 · f900702eaf
parent 6f4aed64f4
commit f900702eaf
1 changed files with 1 additions and 1 deletions
--- a/config.json
+++ b/config.json
@ -1,6 +1,6 @@
 {
  "architectures": [
-    "SegformerForImageSegmentation"
+    "SegformerForSemanticSegmentation"
  ],
  "attention_probs_dropout_prob": 0.0,
  "classifier_dropout_prob": 0.1,