config.json

{
  "downstream": {
    "criterion": {
      "edge_value": 1.0,
      "extended_loss_weight": 0.001,
      "feat_coeff": 10.0
    },
    "feats": {
      "boundary_prediction": false,
      "silence_prediction": true,
      "sum_diphthong": false,
      "unknown_mode": "unk-ign"
    },
    "freeze_hubert": false,
    "model_kwargs": {
      "bidirection": true,
      "dim": [
        1024,
        1024
      ],
      "dropout": [
        0.2,
        0.2
      ],
      "layer_norm": [
        false,
        false
      ],
      "module": "LSTM",
      "proj": [
        false,
        false
      ]
    },
    "model_select": "rnn",
    "neg_threshold": 0.5,
    "phone_head": {
      "activation": "GELU",
      "hiddens": [
        1024
      ]
    },
    "pos_threshold": 0.5,
    "project_dim": 1024
  },
  "hubert_size": "base"
}