-
Notifications
You must be signed in to change notification settings - Fork 1.1k
/
squad_bert_multilingual_freezed_emb.json
70 lines (69 loc) · 2.4 KB
/
squad_bert_multilingual_freezed_emb.json
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
{
"chainer": {
"in": ["context_raw", "question_raw"],
"in_y": ["ans_raw", "ans_raw_start"],
"pipe": [
{
"class_name": "bert_preprocessor",
"vocab_file": "{DOWNLOADS_PATH}/bert_models/multi_cased_L-12_H-768_A-12/vocab.txt",
"do_lower_case": false,
"max_seq_length": 384,
"in": ["question_raw", "context_raw"],
"out": ["bert_features"]
},
{
"class_name": "squad_bert_mapping",
"do_lower_case": false,
"in": ["context_raw", "bert_features"],
"out": ["subtok2chars", "char2subtoks"]
},
{
"class_name": "squad_bert_ans_preprocessor",
"do_lower_case": false,
"in": ["ans_raw", "ans_raw_start","char2subtoks"],
"out": ["ans", "ans_start", "ans_end"]
},
{
"class_name": "squad_bert_model",
"bert_config_file": "{DOWNLOADS_PATH}/bert_models/multi_cased_L-12_H-768_A-12/bert_config.json",
"pretrained_bert": "{DOWNLOADS_PATH}/bert_models/multi_cased_L-12_H-768_A-12/bert_model.ckpt",
"save_path": "{MODELS_PATH}/squad_bert/model_multi_freezed",
"load_path": "{MODELS_PATH}/squad_bert/model_multi_freezed",
"keep_prob": 0.5,
"learning_rate": 2e-05,
"learning_rate_drop_patience": 2,
"learning_rate_drop_div": 2.0,
"in": ["bert_features"],
"in_y": ["ans_start", "ans_end"],
"out": ["ans_start_predicted", "ans_end_predicted", "logits"]
},
{
"class_name": "squad_bert_ans_postprocessor",
"in": ["ans_start_predicted", "ans_end_predicted", "context_raw", "bert_features", "subtok2chars"],
"out": ["ans_predicted", "ans_start_predicted", "ans_end_predicted"]
}
],
"out": ["ans_predicted", "ans_start_predicted", "logits"]
},
"metadata": {
"variables": {
"ROOT_PATH": "~/.deeppavlov",
"DOWNLOADS_PATH": "{ROOT_PATH}/downloads",
"MODELS_PATH": "{ROOT_PATH}/models"
},
"requirements": [
"{DEEPPAVLOV_PATH}/requirements/tf.txt",
"{DEEPPAVLOV_PATH}/requirements/bert_dp.txt"
],
"download": [
{
"url": "http://files.deeppavlov.ai/deeppavlov_data/bert/multi_cased_L-12_H-768_A-12.zip",
"subdir": "{DOWNLOADS_PATH}/bert_models"
},
{
"url": "http://files.deeppavlov.ai/deeppavlov_data/squad_bert_mult_freezed.tar.gz",
"subdir": "{MODELS_PATH}"
}
]
}
}