bdotloh's picture
Upload DistilBertForSequenceClassification
bf28044
raw
history blame
1.97 kB
{
"_name_or_path": "distilbert-base-uncased",
"activation": "gelu",
"architectures": [
"DistilBertForSequenceClassification"
],
"attention_dropout": 0.1,
"dim": 768,
"dropout": 0.1,
"hidden_dim": 3072,
"id2label": {
"0": "afraid",
"1": "angry",
"2": "annoyed",
"3": "anticipating",
"4": "anxious",
"5": "apprehensive",
"6": "ashamed",
"7": "caring",
"8": "confident",
"9": "content",
"10": "devastated",
"11": "disappointed",
"12": "disgusted",
"13": "embarrassed",
"14": "excited",
"15": "faithful",
"16": "furious",
"17": "grateful",
"18": "guilty",
"19": "hopeful",
"20": "impressed",
"21": "jealous",
"22": "joyful",
"23": "lonely",
"24": "nostalgic",
"25": "prepared",
"26": "proud",
"27": "sad",
"28": "sentimental",
"29": "surprised",
"30": "terrified",
"31": "trusting"
},
"initializer_range": 0.02,
"label2id": {
"afraid": 0,
"angry": 1,
"annoyed": 2,
"anticipating": 3,
"anxious": 4,
"apprehensive": 5,
"ashamed": 6,
"caring": 7,
"confident": 8,
"content": 9,
"devastated": 10,
"disappointed": 11,
"disgusted": 12,
"embarrassed": 13,
"excited": 14,
"faithful": 15,
"furious": 16,
"grateful": 17,
"guilty": 18,
"hopeful": 19,
"impressed": 20,
"jealous": 21,
"joyful": 22,
"lonely": 23,
"nostalgic": 24,
"prepared": 25,
"proud": 26,
"sad": 27,
"sentimental": 28,
"surprised": 29,
"terrified": 30,
"trusting": 31
},
"max_position_embeddings": 512,
"model_type": "distilbert",
"n_heads": 12,
"n_layers": 6,
"pad_token_id": 0,
"problem_type": "multi_label_classification",
"qa_dropout": 0.1,
"seq_classif_dropout": 0.2,
"sinusoidal_pos_embds": false,
"tie_weights_": true,
"torch_dtype": "float32",
"transformers_version": "4.22.1",
"vocab_size": 30522
}