From f9c3b1ce7ea1e7795e36d28aa2220654f8de78a4 Mon Sep 17 00:00:00 2001 From: joaogante Date: Tue, 19 Jul 2022 10:39:43 +0000 Subject: [PATCH] Update transformers architecture class (#1) - Update transformers architecture class (8881fd6b4f902d6a4caf297f11a559787746cd53) --- config.json | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/config.json b/config.json index 6490bd3..3cf1f37 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,