From ce902c9df5bbf54f37d1a3012fcb43d307d7b1dd Mon Sep 17 00:00:00 2001 From: ydshieh Date: Thu, 7 Apr 2022 14:27:43 +0200 Subject: [PATCH 1/3] fix QA sample --- src/transformers/utils/doc.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/src/transformers/utils/doc.py b/src/transformers/utils/doc.py index fbdd0f0b2ee0..791d9f6500d0 100644 --- a/src/transformers/utils/doc.py +++ b/src/transformers/utils/doc.py @@ -207,7 +207,9 @@ def _prepare_output_docstrings(output_type, config_class, min_indent=None): ```python >>> # target is "nice puppet" - >>> target_start_index, target_end_index = torch.tensor([14]), torch.tensor([15]) + >>> target_start_index, target_end_index = torch.tensor([{qa_target_start_index}]), torch.tensor( + ... [{qa_target_end_index}] + ... ) >>> outputs = model(**inputs, start_positions=target_start_index, end_positions=target_end_index) >>> loss = outputs.loss @@ -1054,6 +1056,8 @@ def add_code_sample_docstrings( output_type=None, config_class=None, mask="[MASK]", + qa_target_start_index=14, + qa_target_end_index=15, model_cls=None, modality=None, expected_output="", @@ -1078,6 +1082,8 @@ def docstring_decorator(fn): processor_class=processor_class, checkpoint=checkpoint, mask=mask, + qa_target_start_index=qa_target_start_index, + qa_target_end_index=qa_target_end_index, expected_output=expected_output, expected_loss=expected_loss, ) From f33eabfd90365d5ea4374eca5b39ba6b3217b6db Mon Sep 17 00:00:00 2001 From: ydshieh Date: Thu, 7 Apr 2022 16:47:01 +0200 Subject: [PATCH 2/3] use 2 lines --- src/transformers/utils/doc.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/src/transformers/utils/doc.py b/src/transformers/utils/doc.py index 791d9f6500d0..04b3bf0356f4 100644 --- a/src/transformers/utils/doc.py +++ b/src/transformers/utils/doc.py @@ -207,9 +207,8 @@ def _prepare_output_docstrings(output_type, config_class, min_indent=None): ```python >>> # target is "nice puppet" - >>> target_start_index, target_end_index = torch.tensor([{qa_target_start_index}]), torch.tensor( - ... [{qa_target_end_index}] - ... ) + >>> target_start_index = torch.tensor([{qa_target_start_index}]) + >>> target_end_index = torch.tensor([{qa_target_end_index}]) >>> outputs = model(**inputs, start_positions=target_start_index, end_positions=target_end_index) >>> loss = outputs.loss From 94c5797b1be94782c1cb23f49c0701c0247f4af0 Mon Sep 17 00:00:00 2001 From: ydshieh Date: Fri, 8 Apr 2022 14:41:16 +0200 Subject: [PATCH 3/3] For TF_QUESTION_ANSWERING_SAMPLE --- src/transformers/utils/doc.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/transformers/utils/doc.py b/src/transformers/utils/doc.py index 04b3bf0356f4..39508e18d222 100644 --- a/src/transformers/utils/doc.py +++ b/src/transformers/utils/doc.py @@ -668,7 +668,8 @@ def _prepare_output_docstrings(output_type, config_class, min_indent=None): ```python >>> # target is "nice puppet" - >>> target_start_index, target_end_index = tf.constant([14]), tf.constant([15]) + >>> target_start_index = tf.constant([{qa_target_start_index}]) + >>> target_end_index = tf.constant([{qa_target_end_index}]) >>> outputs = model(**inputs, start_positions=target_start_index, end_positions=target_end_index) >>> loss = tf.math.reduce_mean(outputs.loss)