Update transformers architecture class (#1)

- Update transformers architecture class (063777423477ddb24472baec144730aea78cc0c2)
2022-07-19 10:45:11 +00:00 · 2022-07-19 10:45:11 +00:00 · 797fb91281
parent 3c366b609b
commit 797fb91281
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,