Add possibility to maintain full copies of files (#12312)

This commit is contained in:
Sylvain Gugger
2021-06-28 10:02:53 -04:00
committed by GitHub
parent 9490d668d2
commit 57461ac0b4
2 changed files with 30 additions and 0 deletions

View File

@@ -38,6 +38,7 @@ def postprocess_qa_predictions(
null_score_diff_threshold: float = 0.0,
output_dir: Optional[str] = None,
prefix: Optional[str] = None,
is_world_process_zero: bool = True,
):
"""
Post-processes the predictions of a question-answering model to convert them to answers that are substrings of the
@@ -90,6 +91,7 @@ def postprocess_qa_predictions(
scores_diff_json = collections.OrderedDict()
# Logging.
logger.setLevel(logging.INFO if is_world_process_zero else logging.WARN)
logger.info(f"Post-processing {len(examples)} example predictions split into {len(features)} features.")
# Let's loop over all the examples!