Shaltiel commited on
Commit
2b257bd
1 Parent(s): 3e107d9

Update README.md

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