head_type: saplma
feature_extractor:
- name: luh.feature_extractors.basic_hidden_states
  layer_nums: -1
uncertainty_head:
  internal_dim1: 256
  internal_dim2: 128
  internal_dim3: 64