huggingface
/
distilgpt2
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:08 +00:00
parent 51ab088a71
commit b09826e12b
1 changed files with 11 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
config.json
View File

@ -1,28 +1,32 @@
{ {
"_num_labels": 1,
"activation_function": "gelu_new",
"architectures": [ "architectures": [
"GPT2LMHeadModel" "GPT2LMHeadModel"
], ],
"attn_pdrop": 0.1, "attn_pdrop": 0.1,
"bos_token_id": 50256,
"embd_pdrop": 0.1, "embd_pdrop": 0.1,
"finetuning_task": null, "eos_token_id": 50256,
"id2label": {
"0": "LABEL_0"
},
"initializer_range": 0.02, "initializer_range": 0.02,
"label2id": {
"LABEL_0": 0
},
"layer_norm_epsilon": 1e-05, "layer_norm_epsilon": 1e-05,
"model_type": "gpt2",
"n_ctx": 1024, "n_ctx": 1024,
"n_embd": 768, "n_embd": 768,
"n_head": 12, "n_head": 12,
"n_layer": 6, "n_layer": 6,
"n_positions": 1024, "n_positions": 1024,
"num_labels": 1,
"output_attentions": false,
"output_hidden_states": false,
"pruned_heads": {},
"resid_pdrop": 0.1, "resid_pdrop": 0.1,
"summary_activation": null, "summary_activation": null,
"summary_first_dropout": 0.1, "summary_first_dropout": 0.1,
"summary_proj_to_labels": true, "summary_proj_to_labels": true,
"summary_type": "cls_index", "summary_type": "cls_index",
"summary_use_proj": true, "summary_use_proj": true,
"torchscript": false,
"use_bfloat16": false,
"vocab_size": 50257 "vocab_size": 50257
} }