config.yaml

head_type: luh
feature_extractor:
- name: luq.feature_extractors.basic_hidden_states
  layer_nums:
  - -1
uncertainty_head:
  head_dim: 256
  n_layers: 2
  n_heads: 8
  dropout: 0.2
  enable_feature_projection_layer: true
  interim_dim: 256