Team Ai
Modelpublic

Michonas/LLM_objectClassification

sourceHugging Faceupdated 1y agoView on Hugging Face
0likes
config.json53 linesDownload Raw Back to my_trained_model
1{2  "activation": "gelu",3  "architectures": [4    "DistilBertForSequenceClassification"5  ],6  "attention_dropout": 0.1,7  "dim": 768,8  "dropout": 0.1,9  "hidden_dim": 3072,10  "id2label": {11    "0": "LABEL_0",12    "1": "LABEL_1",13    "2": "LABEL_2",14    "3": "LABEL_3",15    "4": "LABEL_4",16    "5": "LABEL_5",17    "6": "LABEL_6",18    "7": "LABEL_7",19    "8": "LABEL_8",20    "9": "LABEL_9",21    "10": "LABEL_10",22    "11": "LABEL_11"23  },24  "initializer_range": 0.02,25  "label2id": {26    "LABEL_0": 0,27    "LABEL_1": 1,28    "LABEL_10": 10,29    "LABEL_11": 11,30    "LABEL_2": 2,31    "LABEL_3": 3,32    "LABEL_4": 4,33    "LABEL_5": 5,34    "LABEL_6": 6,35    "LABEL_7": 7,36    "LABEL_8": 8,37    "LABEL_9": 938  },39  "max_position_embeddings": 512,40  "model_type": "distilbert",41  "n_heads": 12,42  "n_layers": 6,43  "pad_token_id": 0,44  "problem_type": "single_label_classification",45  "qa_dropout": 0.1,46  "seq_classif_dropout": 0.2,47  "sinusoidal_pos_embds": false,48  "tie_weights_": true,49  "torch_dtype": "float32",50  "transformers_version": "4.52.4",51  "vocab_size": 3052252}53