Dreamy0 commited on
Commit
383adb7
·
verified ·
1 Parent(s): d76f72d

Update config.json

Browse files
Files changed (1) hide show
  1. 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],