Team Ai
Modelpublic

WilliamStar/sequence_classification_model

sourceHugging Faceupdated 2y agoView on Hugging Face
0likes12downloads
config.json39 linesDownload Raw Back to root
1{2  "_name_or_path": "squeezebert/squeezebert-uncased",3  "architectures": [4    "SqueezeBertForSequenceClassification"5  ],6  "attention_probs_dropout_prob": 0.1,7  "embedding_size": 768,8  "hidden_act": "gelu",9  "hidden_dropout_prob": 0.1,10  "hidden_size": 768,11  "id2label": {12    "0": "NEGATIVE",13    "1": "POSITIVE"14  },15  "initializer_range": 0.02,16  "intermediate_groups": 4,17  "intermediate_size": 3072,18  "k_groups": 4,19  "label2id": {20    "NEGATIVE": 0,21    "POSITIVE": 122  },23  "layer_norm_eps": 1e-12,24  "max_position_embeddings": 512,25  "model_type": "squeezebert",26  "num_attention_heads": 12,27  "num_hidden_layers": 12,28  "output_groups": 4,29  "pad_token_id": 0,30  "post_attention_groups": 1,31  "problem_type": "single_label_classification",32  "q_groups": 4,33  "torch_dtype": "float32",34  "transformers_version": "4.30.0",35  "type_vocab_size": 2,36  "v_groups": 4,37  "vocab_size": 3052838}39