Update model name

2021-10-15 10:02:54 +00:00 · 2021-10-15 10:02:54 +00:00 · 57e4a51cad
parent 77fa31aeb8
commit 57e4a51cad
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,