Add TF2 XLM-RoBERTa model

This commit is contained in:
Julien Plu
2020-01-08 14:22:56 +01:00
committed by Lysandre Debut
parent 9d87eafd11
commit 64ca855617
5 changed files with 304 additions and 8 deletions

View File

@@ -52,7 +52,6 @@ from utils_squad import (
write_predictions,
write_predictions_extended,
)
# The follwing import is the official SQuAD evaluation script (2.0).
# You can remove it from the dependencies if you are using this script outside of the library
# We've added it here for automated tests (see examples/test_examples.py file)

View File

@@ -21,7 +21,6 @@ import logging
import math
from transformers.tokenization_bert import BasicTokenizer, whitespace_tokenize
# Required by XLNet evaluation method to compute optimal threshold (see write_predictions_extended() method)
from utils_squad_evaluate import find_all_best_thresh_v2, get_raw_scores, make_qid_to_has_ans