microsoft
/
DialoGPT-medium
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 16:02:37 +00:00
parent b86761883d
commit a901f37c2a
1 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
config.json
View File

@ -1,8 +1,12 @@
{ {
"activation_function": "gelu_new",
"architectures": [ "architectures": [
"GPT2LMHeadModel" "GPT2LMHeadModel"
], ],
"attn_pdrop": 0.1,
"bos_token_id": 50256, "bos_token_id": 50256,
"embd_pdrop": 0.1,
"eos_token_id": 50256,
"initializer_range": 0.02, "initializer_range": 0.02,
"layer_norm_epsilon": 1e-05, "layer_norm_epsilon": 1e-05,
"model_type": "gpt2", "model_type": "gpt2",
@ -11,5 +15,11 @@
"n_head": 16, "n_head": 16,
"n_layer": 24, "n_layer": 24,
"n_positions": 1024, "n_positions": 1024,
"resid_pdrop": 0.1,
"summary_activation": null,
"summary_first_dropout": 0.1,
"summary_proj_to_labels": true,
"summary_type": "cls_index",
"summary_use_proj": true,
"vocab_size": 50257 "vocab_size": 50257
} }