ankur310794's picture
add model
5a9d130
{
"_name_or_path": "output",
"activation": "gelu",
"architectures": [
"DistilBertForSequenceClassification"
],
"attention_dropout": 0.1,
"dim": 768,
"dropout": 0.1,
"hidden_dim": 3072,
"id2label": {
"0": "occupation",
"1": "founder",
"2": "owned by",
"3": "located in the administrative territorial entity",
"4": "genre",
"5": "headquarters location",
"6": "country",
"7": "creator",
"8": "performer",
"9": "manufacturer",
"10": "place of birth",
"11": "place of death",
"12": "spouse",
"13": "record label",
"14": "location",
"15": "capital",
"16": "child",
"17": "language of work or name",
"18": "position played on team / speciality",
"19": "country of origin",
"20": "author",
"21": "educated at",
"22": "location of formation",
"23": "notable work"
},
"initializer_range": 0.02,
"label2id": {
"author": 20,
"capital": 15,
"child": 16,
"country": 6,
"country of origin": 19,
"creator": 7,
"educated at": 21,
"founder": 1,
"genre": 4,
"headquarters location": 5,
"language of work or name": 17,
"located in the administrative territorial entity": 3,
"location": 14,
"location of formation": 22,
"manufacturer": 9,
"notable work": 23,
"occupation": 0,
"owned by": 2,
"performer": 8,
"place of birth": 10,
"place of death": 11,
"position played on team / speciality": 18,
"record label": 13,
"spouse": 12
},
"max_position_embeddings": 512,
"model_type": "distilbert",
"n_heads": 12,
"n_layers": 6,
"output_past": true,
"pad_token_id": 0,
"qa_dropout": 0.1,
"seq_classif_dropout": 0.2,
"sinusoidal_pos_embds": false,
"tie_weights_": true,
"transformers_version": "4.12.2",
"vocab_size": 28996
}