Update config.json
Browse files- config.json +1 -1
config.json
CHANGED
@@ -13,7 +13,7 @@
|
|
13 |
"hidden_size": 768,
|
14 |
"image_size": 224,
|
15 |
"in_channels": 3,
|
16 |
-
"model_type": "maskformer-swin"
|
17 |
"num_heads": [3, 6, 12, 24],
|
18 |
"out_features": ["stage1", "stage2", "stage3", "stage4"],
|
19 |
"out_indices": [1, 2, 3, 4],
|
|
|
13 |
"hidden_size": 768,
|
14 |
"image_size": 224,
|
15 |
"in_channels": 3,
|
16 |
+
"model_type": "swin", // Changed from "maskformer-swin" to "swin"
|
17 |
"num_heads": [3, 6, 12, 24],
|
18 |
"out_features": ["stage1", "stage2", "stage3", "stage4"],
|
19 |
"out_indices": [1, 2, 3, 4],
|