Update README.md

This commit is contained in:
Baptiste Doyen 2021-03-25 15:08:19 +00:00 committed by huggingface-web
parent 2a02222200
commit 7848658c7e
1 changed files with 2 additions and 1 deletions

View File

@ -46,7 +46,7 @@ classifier(sequence, candidate_labels, hypothesis_template=hypothesis_template)
``` ```
- As a premise/hypothesis checker : <br> - As a premise/hypothesis checker : <br>
The idea is here to compute a probability of the form \\(P(premise|hypothesis)\\) The idea is here to compute a probability of the form \\\\(P(premise|hypothesis)\\\\)
```python ```python
# load model and tokenizer # load model and tokenizer
@ -86,6 +86,7 @@ Main training parameters :
- ```num_train_epochs = 4``` - ```num_train_epochs = 4```
- ```batch_size = 12``` (limited by GPU-memory) - ```batch_size = 12``` (limited by GPU-memory)
- ```weight_decay = 0.01``` - ```weight_decay = 0.01```
- ```metric_for_best_model = "eval_accuracy"```
## Eval results ## Eval results