dmrau commited on
Commit
bb01858
·
1 Parent(s): 075203f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +13 -1
README.md CHANGED
@@ -2,8 +2,20 @@
2
  license: afl-3.0
3
  ---
4
 
 
 
 
 
 
 
 
 
 
 
 
 
5
  <strong>Example on how to load and use BOW-BERT: <strong>
6
- ```
7
  from transformers import AutoModelForSequenceClassification, AutoTokenizer
8
 
9
  # load model
 
2
  license: afl-3.0
3
  ---
4
 
5
+
6
+ This model was trained with the [Rankers Library](`https://github.com/davidmrau/rankers`) please check https://github.com/davidmrau/rankers#Evaluation to learn how to use the model.
7
+
8
+ This Cross Encoder Model was trained on the official MS MARCO training triples using the following training parameters:
9
+
10
+ - batch size: 128
11
+
12
+ - learning rate 3e-6
13
+
14
+ - Optimizer: Adam
15
+
16
+
17
  <strong>Example on how to load and use BOW-BERT: <strong>
18
+ ```python
19
  from transformers import AutoModelForSequenceClassification, AutoTokenizer
20
 
21
  # load model