Update transformers architecture class (#1)

- Update transformers architecture class (8881fd6b4f902d6a4caf297f11a559787746cd53)
2022-07-19 10:39:43 +00:00 · 2022-07-19 10:39:43 +00:00 · f9c3b1ce7e
parent 7993b42804
commit f9c3b1ce7e
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,