office-character / config.json
kearney's picture
End of training
c3c5dbe
raw
history blame
No virus
1.77 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": "Andy",
"1": "Angela",
"2": "Charles",
"3": "Clark",
"4": "Creed",
"5": "Darryl",
"6": "David",
"7": "David Wallace",
"8": "Deangelo",
"9": "Dwight",
"10": "Erin",
"11": "Gabe",
"12": "Holly",
"13": "Jan",
"14": "Jim",
"15": "Jo",
"16": "Karen",
"17": "Kelly",
"18": "Kevin",
"19": "Meredith",
"20": "Michael",
"21": "Nellie",
"22": "Oscar",
"23": "Pam",
"24": "Pete",
"25": "Phyllis",
"26": "Robert",
"27": "Roy",
"28": "Ryan",
"29": "Stanley",
"30": "Toby"
},
"initializer_range": 0.02,
"label2id": {
"Andy": 0,
"Angela": 1,
"Charles": 2,
"Clark": 3,
"Creed": 4,
"Darryl": 5,
"David": 6,
"David Wallace": 7,
"Deangelo": 8,
"Dwight": 9,
"Erin": 10,
"Gabe": 11,
"Holly": 12,
"Jan": 13,
"Jim": 14,
"Jo": 15,
"Karen": 16,
"Kelly": 17,
"Kevin": 18,
"Meredith": 19,
"Michael": 20,
"Nellie": 21,
"Oscar": 22,
"Pam": 23,
"Pete": 24,
"Phyllis": 25,
"Robert": 26,
"Roy": 27,
"Ryan": 28,
"Stanley": 29,
"Toby": 30
},
"max_position_embeddings": 512,
"model_type": "distilbert",
"n_heads": 12,
"n_layers": 6,
"pad_token_id": 0,
"problem_type": "single_label_classification",
"qa_dropout": 0.1,
"seq_classif_dropout": 0.2,
"sinusoidal_pos_embds": false,
"tie_weights_": true,
"torch_dtype": "float32",
"transformers_version": "4.32.1",
"vocab_size": 30522
}