office-character / config.json
kearney's picture
Training in progress, step 500
3e9fa4d
raw
history blame
2.25 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": "Abby",
"1": "Andy",
"2": "Angela",
"3": "Billy Merchant",
"4": "Captain Jack",
"5": "Carol",
"6": "Christian",
"7": "Craig",
"8": "Creed",
"9": "Darryl",
"10": "David",
"11": "Dwight",
"12": "Hannah",
"13": "Improv Teacher",
"14": "Jake",
"15": "Jan",
"16": "Jim",
"17": "Josh",
"18": "Karen",
"19": "Katy",
"20": "Kelly",
"21": "Kevin",
"22": "Martin",
"23": "Mary-Beth",
"24": "Melissa",
"25": "Meredith",
"26": "Michael",
"27": "Mr. Brown",
"28": "Oscar",
"29": "Pam",
"30": "Pam's Mom",
"31": "Photographer",
"32": "Phyllis",
"33": "Roy",
"34": "Ryan",
"35": "Sadiq (IT guy)",
"36": "Second Cindy",
"37": "Stanley",
"38": "Toby",
"39": "Todd Packer",
"40": "Tony"
},
"initializer_range": 0.02,
"label2id": {
"Abby": 0,
"Andy": 1,
"Angela": 2,
"Billy Merchant": 3,
"Captain Jack": 4,
"Carol": 5,
"Christian": 6,
"Craig": 7,
"Creed": 8,
"Darryl": 9,
"David": 10,
"Dwight": 11,
"Hannah": 12,
"Improv Teacher": 13,
"Jake": 14,
"Jan": 15,
"Jim": 16,
"Josh": 17,
"Karen": 18,
"Katy": 19,
"Kelly": 20,
"Kevin": 21,
"Martin": 22,
"Mary-Beth": 23,
"Melissa": 24,
"Meredith": 25,
"Michael": 26,
"Mr. Brown": 27,
"Oscar": 28,
"Pam": 29,
"Pam's Mom": 30,
"Photographer": 31,
"Phyllis": 32,
"Roy": 33,
"Ryan": 34,
"Sadiq (IT guy)": 35,
"Second Cindy": 36,
"Stanley": 37,
"Toby": 38,
"Todd Packer": 39,
"Tony": 40
},
"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
}