From 043b66a442dca5da30647ca2b6cda32cc3ece049 Mon Sep 17 00:00:00 2001 From: Niels Rogge Date: Fri, 15 Oct 2021 10:04:05 +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 fd47473..d7a69e5 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,