Team Ai
Modelpublic

MITLL/LADI-v2-classifier-large-reference

sourceHugging Facemitupdated 2y agoView on Hugging Face
1likes13downloads
config.json88 linesDownload Raw Back to root
1{2  "_name_or_path": "/home/gridsan/groups/CAP_shared/finetuned_models/model_d0fCfn9I/run_20240528-155859_q8n2zbS/epoch_049",3  "architectures": [4    "Swinv2ForImageClassification"5  ],6  "attention_probs_dropout_prob": 0.0,7  "depths": [8    2,9    2,10    18,11    212  ],13  "drop_path_rate": 0.1,14  "embed_dim": 192,15  "encoder_stride": 32,16  "hidden_act": "gelu",17  "hidden_dropout_prob": 0.0,18  "hidden_size": 1536,19  "id2label": {20    "0": "bridges_any",21    "1": "buildings_any",22    "2": "buildings_affected_or_greater",23    "3": "buildings_minor_or_greater",24    "4": "debris_any",25    "5": "flooding_any",26    "6": "flooding_structures",27    "7": "roads_any",28    "8": "roads_damage",29    "9": "trees_any",30    "10": "trees_damage",31    "11": "water_any"32  },33  "image_size": 256,34  "initializer_range": 0.02,35  "label2id": {36    "bridges_any": "0",37    "buildings_affected_or_greater": "2",38    "buildings_any": "1",39    "buildings_minor_or_greater": "3",40    "debris_any": "4",41    "flooding_any": "5",42    "flooding_structures": "6",43    "roads_any": "7",44    "roads_damage": "8",45    "trees_any": "9",46    "trees_damage": "10",47    "water_any": "11"48  },49  "layer_norm_eps": 1e-05,50  "mlp_ratio": 4.0,51  "model_type": "swinv2",52  "num_channels": 3,53  "num_heads": [54    6,55    12,56    24,57    4858  ],59  "num_layers": 4,60  "out_features": [61    "stage4"62  ],63  "out_indices": [64    465  ],66  "patch_size": 4,67  "path_norm": true,68  "pretrained_window_sizes": [69    12,70    12,71    12,72    673  ],74  "problem_type": "multi_label_classification",75  "qkv_bias": true,76  "stage_names": [77    "stem",78    "stage1",79    "stage2",80    "stage3",81    "stage4"82  ],83  "torch_dtype": "float32",84  "transformers_version": "4.37.2",85  "use_absolute_embeddings": false,86  "window_size": 1687}88