rediska0123 commited on
Commit
dce12d3
·
1 Parent(s): 8d775cf

Uploading model

Browse files
Files changed (2) hide show
  1. config.yaml +23 -0
  2. weights.pth +3 -0
config.yaml ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ feature_extractor:
2
+ - name: luq.feature_extractors.basic_hidden_states
3
+ layer_nums:
4
+ - -1
5
+ - -5
6
+ - -16
7
+ - name: luq.feature_extractors.basic_attention
8
+ layer_nums:
9
+ - -1
10
+ - -5
11
+ - -10
12
+ - -16
13
+ - -20
14
+ - -25
15
+ - name: luq.feature_extractors.token_probabilities
16
+ top_n: 3
17
+ uncertainty_head:
18
+ head_dim: 4096
19
+ n_layers: 1
20
+ n_heads: 32
21
+ dropout: 0.1
22
+ enable_feature_projection_layer: true
23
+ interim_dim: 1024
weights.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3a379145a16eb4c398fc035e961de871af1f058ee093e6d76365fb727bc16f3d
3
+ size 2657278902