Shaltiel commited on
Commit
8f97bab
1 Parent(s): 4cbe83a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -0
README.md CHANGED
@@ -21,6 +21,9 @@ from transformers import AutoTokenizer, BertForMaskedLM
21
 
22
  tokenizer = AutoTokenizer.from_pretrained('dicta-il/BEREL_2.0')
23
  model = BertForMaskedLM.from_pretrained('dicta-il/BEREL_2.0')
 
 
 
24
  ```
25
 
26
  > NOTE: This code will **not** work and provide bad results if you use `BertTokenizer`. Please use `AutoTokenizer` or `BertTokenizerFast`.
 
21
 
22
  tokenizer = AutoTokenizer.from_pretrained('dicta-il/BEREL_2.0')
23
  model = BertForMaskedLM.from_pretrained('dicta-il/BEREL_2.0')
24
+
25
+ # for evaluation, disable dropout
26
+ model.eval()
27
  ```
28
 
29
  > NOTE: This code will **not** work and provide bad results if you use `BertTokenizer`. Please use `AutoTokenizer` or `BertTokenizerFast`.