Artem's picture
Upload UncertaintyHead model and config
87f1943
raw
history blame contribute delete
435 Bytes
head_type: luh
feature_extractor:
- name: luq.feature_extractors.basic_hidden_states
layer_nums:
- -1
- -7
- -16
- -20
- name: luq.feature_extractors.basic_attention
layer_nums:
- -1
- -5
- -10
- -16
- -20
- -25
- name: luq.feature_extractors.token_probabilities
top_n: 5
uncertainty_head:
head_dim: 4096
n_layers: 1
n_heads: 64
dropout: 0.1
enable_feature_projection_layer: true
interim_dim: 512