"doc/git@developer.sourcefind.cn:wangsen/paddle_dbnet.git" did not exist on "c249ba9feac719ed34c20f6a0edcf90686d1cac3"
Unverified Commit 1889e96c authored by Patrick von Platen's avatar Patrick von Platen Committed by GitHub
Browse files

fix QA example for PT (#6890)

parent d822ab63
...@@ -303,14 +303,15 @@ PT_QUESTION_ANSWERING_SAMPLE = r""" ...@@ -303,14 +303,15 @@ PT_QUESTION_ANSWERING_SAMPLE = r"""
>>> tokenizer = {tokenizer_class}.from_pretrained('{checkpoint}') >>> tokenizer = {tokenizer_class}.from_pretrained('{checkpoint}')
>>> model = {model_class}.from_pretrained('{checkpoint}', return_dict=True) >>> 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]) >>> start_positions = torch.tensor([1])
>>> end_positions = torch.tensor([3]) >>> end_positions = torch.tensor([3])
>>> outputs = model(**inputs, start_positions=start_positions, end_positions=end_positions) >>> outputs = model(**inputs, start_positions=start_positions, end_positions=end_positions)
>>> loss = outputs.loss >>> loss = outputs.loss
>>> start_scores = outputs.start_scores >>> start_scores = outputs.start_logits
>>> end_scores = outputs.end_scores >>> end_scores = outputs.end_logits
""" """
PT_SEQUENCE_CLASSIFICATION_SAMPLE = r""" PT_SEQUENCE_CLASSIFICATION_SAMPLE = r"""
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment