dream

1
{
2
    "chainer": {
3
      "in": ["uttr", "rel_list"],
4
      "pipe": [
5
        {
6
          "class_name": "torch_transformers_preprocessor",
7
          "vocab_file": "{TRANSFORMER}",
8
          "do_lower_case": true,
9
          "max_seq_length": 128,
10
          "in": ["uttr", "rel_list"],
11
          "out": ["bert_features"]
12
        },
13
        {
14
          "class_name": "torch_transformers_classifier",
15
          "n_classes": 2,
16
          "return_probas": true,
17
          "pretrained_bert": "{TRANSFORMER}",
18
          "save_path": "{MODEL_PATH}/model_uncased",
19
          "load_path": "{MODEL_PATH}/model_uncased",
20
          "optimizer": "AdamW",
21
          "optimizer_parameters": {"lr": 1e-05},
22
          "learning_rate_drop_patience": 5,
23
          "learning_rate_drop_div": 2.0,
24
          "in": ["bert_features"],
25
          "out": ["y_pred_probas"]
26
        }
27
      ],
28
      "out": ["y_pred_probas"]
29
    },
30
    "metadata": {
31
      "variables": {
32
        "ROOT_PATH": "~/.deeppavlov",
33
        "DOWNLOADS_PATH": "{ROOT_PATH}/downloads",
34
        "TRANSFORMER": "google/bert_uncased_L-4_H-512_A-8",
35
        "MODEL_PATH": "{ROOT_PATH}/models/classifiers/rel_ranking_prex"
36
      },
37
      "download": [
38
        {
39
          "url": "http://files.deeppavlov.ai/deeppavlov_data/generative_ie/rel_classifier_uncased.tar.gz",
40
          "subdir": "{MODEL_PATH}"
41
        }
42
      ]
43
    }
44
}