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],
|