config.json

{
  "model": {
    "use_lora": false,
    "lora_r": 64,
    "lora_alpha": 128,
    "lora_dropout": 0.05,
    "lora_target_modules": [
      "query",
      "key",
      "value",
      "output.dense"
    ],
    "arc_hidden_dim": 256,
    "rel_hidden_dim": 128,
    "char_embed_dim": 300,
    "max_char_len": 50,
    "dropout": 0.1,
    "lemma_input_proj_dim": 32,
    "lemma_input_proj_dropout": 0.25,
    "lemma_cnn_filters": [
      256,
      256,
      256
    ],
    "lemma_cnn_kernel_sizes": [
      3,
      3,
      3,
      1
    ],
    "lemma_cnn_dilations": [
      1,
      2,
      4,
      1
    ],
    "lemma_cnn_paddings": [
      1,
      2,
      4,
      0
    ],
    "morpho_weight": 1.0,
    "feats_weight": 3.0,
    "arc_weight": 1.0,
    "rel_weight": 1.0,
    "lemma_weight": 1.5,
    "use_uncertainty_weighting": false,
    "morpho_to_lemma_proj_dim": 64,
    "lemma_beam_width": 5,
    "use_lemma_beam_search": true,
    "use_morpho_constraints": true,
    "base_model": "FacebookAI/xlm-roberta-base",
    "num_upos": 19,
    "num_xpos": 348,
    "feat_vocab_sizes": {
      "Abbr": 3,
      "AdpType": 3,
      "Aspect": 4,
      "Case": 5,
      "ConjType": 3,
      "Definite": 6,
      "ExtPos": 5,
      "Foreign": 3,
      "Gender": 4,
      "Mood": 6,
      "NumForm": 4,
      "NumValue": 5,
      "Number": 5,
      "Person": 5,
      "Polarity": 3,
      "PronType": 6,
      "VerbForm": 3,
      "Voice": 4
    },
    "feat_categories": [
      "Abbr",
      "AdpType",
      "Aspect",
      "Case",
      "ConjType",
      "Definite",
      "ExtPos",
      "Foreign",
      "Gender",
      "Mood",
      "NumForm",
      "NumValue",
      "Number",
      "Person",
      "Polarity",
      "PronType",
      "VerbForm",
      "Voice"
    ],
    "num_deprels": 39,
    "char_vocab_size": 109
  },
  "max_length": 512
}