CoolFace
Modelpublic

SharonTudi/DIALOGUE_four_model

sourceHugging Faceapache-2.0updated 3y agoView on Hugging Face
0likes4downloads
config.json38 linesDownload Raw Back to root
1{2  "_name_or_path": "distilbert-base-uncased",3  "activation": "gelu",4  "architectures": [5    "DistilBertForSequenceClassification"6  ],7  "attention_dropout": 0.1,8  "dim": 768,9  "dropout": 0.1,10  "hidden_dim": 3072,11  "id2label": {12    "0": "Hospital-Inform",13    "1": "Hospital-Request",14    "2": "general-bye",15    "3": "general-thank"16  },17  "initializer_range": 0.02,18  "label2id": {19    "LABEL_0": 0,20    "LABEL_1": 1,21    "LABEL_2": 2,22    "LABEL_3": 323  },24  "max_position_embeddings": 512,25  "model_type": "distilbert",26  "n_heads": 12,27  "n_layers": 6,28  "pad_token_id": 0,29  "problem_type": "single_label_classification",30  "qa_dropout": 0.1,31  "seq_classif_dropout": 0.2,32  "sinusoidal_pos_embds": false,33  "tie_weights_": true,34  "torch_dtype": "float32",35  "transformers_version": "4.36.2",36  "vocab_size": 3052237}38