From 57e4a51cadb13294834716f1b71901ffb49dc828 Mon Sep 17 00:00:00 2001
From: Niels Rogge <niels.rogge1@gmail.com>
Date: Fri, 15 Oct 2021 10:02:54 +0000
Subject: [PATCH] Update model name

---
 config.json | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/config.json b/config.json
index 8c022d8..ba9e667 100644
--- a/config.json
+++ b/config.json
@@ -1,6 +1,6 @@
 {
   "architectures": [
-    "SegformerForImageSegmentation"
+    "SegformerForSemanticSegmentation"
   ],
   "attention_probs_dropout_prob": 0.0,
   "classifier_dropout_prob": 0.1,