From 797fb91281ebd7ff9107e768fa676484803ac419 Mon Sep 17 00:00:00 2001 From: joaogante Date: Tue, 19 Jul 2022 10:45:11 +0000 Subject: [PATCH] Update transformers architecture class (#1) - Update transformers architecture class (063777423477ddb24472baec144730aea78cc0c2) --- config.json | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/config.json b/config.json index 58f4073..a29f1b2 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,