huggingface
/
distilbert-base-uncased-finetuned-sst-2-english
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update config.json

This commit is contained in:
system 2020-04-24 15:58:05 +00:00
parent d9fd1c824e
commit 9e7b6fe742
1 changed files with 16 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
config.json
View File

@ -1,36 +1,31 @@
{ {
"activation": "gelu",
"architectures": [ "architectures": [
"DistilBertForSequenceClassification" "DistilBertForSequenceClassification"
], ],
"activation": "gelu",
"attention_dropout": 0.1, "attention_dropout": 0.1,
"dim": 768, "dim": 768,
"dropout": 0.1, "dropout": 0.1,
"finetuning_task": "sst-2", "finetuning_task": "sst-2",
"hidden_dim": 3072, "hidden_dim": 3072,
"id2label": {
"0": "NEGATIVE",
"1": "POSITIVE"
},
"initializer_range": 0.02, "initializer_range": 0.02,
"is_decoder": false,
"max_position_embeddings": 512,
"n_heads": 12,
"n_layers": 6,
"num_labels": 2,
"output_attentions": false,
"output_hidden_states": false,
"output_past": true,
"pruned_heads": {},
"qa_dropout": 0.1,
"seq_classif_dropout": 0.2,
"sinusoidal_pos_embds": false,
"tie_weights_": true,
"torchscript": false,
"use_bfloat16": false,
"vocab_size": 30522,
"label2id": { "label2id": {
"NEGATIVE": 0, "NEGATIVE": 0,
"POSITIVE": 1 "POSITIVE": 1
}, },
"id2label": { "max_position_embeddings": 512,
"0": "NEGATIVE", "model_type": "distilbert",
"1": "POSITIVE" "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,
"vocab_size": 30522
} }