Upload BitForImageClassification

config.json

@@ -0,0 +1,74 @@
+{
+  "_name_or_path": "../models/LADI-v2-classifier-small",
+  "architectures": [
+    "BitForImageClassification"
+  ],
+  "conv_layer": "std_conv",
+  "depths": [
+    3,
+    4,
+    6,
+    3
+  ],
+  "drop_path_rate": 0.0,
+  "embedding_dynamic_padding": false,
+  "embedding_size": 64,
+  "global_padding": null,
+  "hidden_act": "relu",
+  "hidden_sizes": [
+    256,
+    512,
+    1024,
+    2048
+  ],
+  "id2label": {
+    "0": "bridges_any",
+    "1": "buildings_any",
+    "2": "buildings_affected_or_greater",
+    "3": "buildings_minor_or_greater",
+    "4": "debris_any",
+    "5": "flooding_any",
+    "6": "flooding_structures",
+    "7": "roads_any",
+    "8": "roads_damage",
+    "9": "trees_any",
+    "10": "trees_damage",
+    "11": "water_any"
+  },
+  "label2id": {
+    "bridges_any": "0",
+    "buildings_affected_or_greater": "2",
+    "buildings_any": "1",
+    "buildings_minor_or_greater": "3",
+    "debris_any": "4",
+    "flooding_any": "5",
+    "flooding_structures": "6",
+    "roads_any": "7",
+    "roads_damage": "8",
+    "trees_any": "9",
+    "trees_damage": "10",
+    "water_any": "11"
+  },
+  "layer_type": "preactivation",
+  "model_type": "bit",
+  "num_channels": 3,
+  "num_groups": 32,
+  "out_features": [
+    "stage4"
+  ],
+  "out_indices": [
+    4
+  ],
+  "output_stride": 32,
+  "problem_type": "multi_label_classification",
+  "stage_names": [
+    "stem",
+    "stage1",
+    "stage2",
+    "stage3",
+    "stage4"
+  ],
+  "torch_dtype": "float32",
+  "transformers_version": "4.37.2",
+  "width_factor": 1
+}

model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5ff777c9d1580eb5bd2f03d0dc5318bdfa77876d49ee1cda14ed70d9c3f47e76
+size 94116688