Upload config

config.json

@@ -1,9 +1,6 @@
 {
-  "_name_or_path": "ksmcg/fan_small_12_p4_hybrid",
   "act_layer": null,
-  "architectures": [
-    "FANForImageClassification"
-  ],
+  "align_corners": false,
   "attn_drop_rate": 0.0,
   "backbone": "hybrid",
   "c_head_num": null,
@@ -22,6 +19,9 @@
   "channels": 256,
   "cls_attn_layers": 2,
   "decoder_hidden_size": 768,
+  "decoder_params": {
+    "embed_dim": 768
+  },
   "depth": 10,
   "depths": [
     3,
@@ -53,8 +53,8 @@
   "in_channels": [
     128,
     256,
-    480,
-    480
+    384,
+    384
   ],
   "in_chans": 3,
   "in_index": [
@@ -64,12 +64,21 @@
     3
   ],
   "initializer_range": 1.0,
+  "loss_decode": {
+    "loss_weight": 1.0,
+    "type": "CrossEntropyLoss",
+    "use_sigmoid": false
+  },
   "mlp_ratio": 4.0,
   "model_type": "fan",
+  "norm_cfg": {
+    "requires_grad": true,
+    "type": "SyncBN"
+  },
   "norm_layer": null,
-  "num_classes": 1000,
+  "num_classes": 19,
   "num_heads": 8,
-  "out_index": -1,
+  "out_index": 9,
   "patch_size": 16,
   "qkv_bias": true,
   "reshape_last_stage": false,
@@ -89,8 +98,8 @@
     1
   ],
   "tokens_norm": true,
-  "torch_dtype": "float32",
   "transformers_version": "4.22.0.dev0",
+  "type": "SegFormerHead",
   "use_checkpoint": false,
   "use_head": false,
   "use_pos_embed": true