Artem's picture
Upload UncertaintyHead model and config
061d71c
raw
history blame
185 Bytes
head_type: saplma
feature_extractor:
- name: luq.feature_extractors.basic_hidden_states
layer_nums: -1
uncertainty_head:
internal_dim1: 256
internal_dim2: 128
internal_dim3: 64