Update README.md
This commit is contained in:
parent
2a02222200
commit
7848658c7e
|
@ -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
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue