Update model name

2021-10-15 10:04:05 +00:00 · 2021-10-15 10:04:05 +00:00 · 043b66a442
parent 55c278c961
commit 043b66a442
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,