From 7848658c7ebe2590c552bbf24fa4539d8461a746 Mon Sep 17 00:00:00 2001 From: Baptiste Doyen Date: Thu, 25 Mar 2021 15:08:19 +0000 Subject: [PATCH] Update README.md --- README.md | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index 3563fe8..df7b9ef 100644 --- a/README.md +++ b/README.md @@ -46,7 +46,7 @@ classifier(sequence, candidate_labels, hypothesis_template=hypothesis_template) ``` - As a premise/hypothesis checker :
-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 # load model and tokenizer @@ -86,6 +86,7 @@ Main training parameters : - ```num_train_epochs = 4``` - ```batch_size = 12``` (limited by GPU-memory) - ```weight_decay = 0.01``` +- ```metric_for_best_model = "eval_accuracy"``` ## Eval results