Andrea Ponti commited on
Commit
ae73393
1 Parent(s): e13d785

Add pooling layer

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": 768,
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
+ }