bwang0911 commited on
Commit
db4bfec
1 Parent(s): 2557c60

feat: add pooling config for sbert

Browse files
Files changed (1) hide show
  1. 1_Pooling/config.json +7 -0
1_Pooling/config.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "word_embedding_dimension": 1024,
3
+ "pooling_mode_cls_token": false,
4
+ "pooling_mode_mean_tokens": true,
5
+ "pooling_mode_max_tokens": false,
6
+ "pooling_mode_mean_sqrt_len_tokens": false
7
+ }