Add model

config.json

@@ -4,12 +4,18 @@
   ],
   "attention_probs_dropout_prob": 0.0,
   "auxiliary_loss_weight": 0.4,
-  "channels": 256,
+  "backbone_out_indices": [
+    5,
+    11,
+    17,
+    23
+  ],
+  "fusion_hidden_size": 256,
+  "head_in_index": -1,
   "hidden_act": "gelu",
   "hidden_dropout_prob": 0.0,
   "hidden_size": 1024,
   "image_size": 384,
-  "in_index": -1,
   "initializer_range": 0.02,
   "intermediate_size": 4096,
   "layer_norm_eps": 1e-12,
@@ -23,12 +29,6 @@
   "num_attention_heads": 16,
   "num_channels": 3,
   "num_hidden_layers": 24,
-  "out_indices": [
-    5,
-    11,
-    17,
-    23
-  ],
   "patch_size": 16,
   "qkv_bias": true,
   "readout_type": "project",
@@ -43,5 +43,5 @@
   "torch_dtype": "float32",
   "transformers_version": "4.18.0.dev0",
   "use_auxiliary_head": true,
-  "use_batch_norm": false
+  "use_batch_norm_in_fusion_residual": false
 }