From 1889e96c8c278a88cb55339e845800e1799f60da Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Wed, 2 Sep 2020 09:53:09 +0200 Subject: [PATCH] fix QA example for PT (#6890) --- src/transformers/file_utils.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/src/transformers/file_utils.py b/src/transformers/file_utils.py index 152e6cae68..382211dca6 100644 --- a/src/transformers/file_utils.py +++ b/src/transformers/file_utils.py @@ -303,14 +303,15 @@ PT_QUESTION_ANSWERING_SAMPLE = r""" >>> tokenizer = {tokenizer_class}.from_pretrained('{checkpoint}') >>> model = {model_class}.from_pretrained('{checkpoint}', return_dict=True) - >>> inputs = tokenizer("Hello, my dog is cute", return_tensors="pt") + >>> question, text = "Who was Jim Henson?", "Jim Henson was a nice puppet" + >>> inputs = tokenizer(question, text, return_tensors='pt') >>> start_positions = torch.tensor([1]) >>> end_positions = torch.tensor([3]) >>> outputs = model(**inputs, start_positions=start_positions, end_positions=end_positions) >>> loss = outputs.loss - >>> start_scores = outputs.start_scores - >>> end_scores = outputs.end_scores + >>> start_scores = outputs.start_logits + >>> end_scores = outputs.end_logits """ PT_SEQUENCE_CLASSIFICATION_SAMPLE = r"""