From 5a6826ca77e3272a224cd44e4f60df7a7b2e3922 Mon Sep 17 00:00:00 2001 From: mymusise Date: Tue, 27 Oct 2020 11:13:22 +0800 Subject: [PATCH 1/2] fix doc bug Signed-off-by: mymusise --- src/transformers/file_utils.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/src/transformers/file_utils.py b/src/transformers/file_utils.py index 23283b085663..e51d9f827e58 100644 --- a/src/transformers/file_utils.py +++ b/src/transformers/file_utils.py @@ -651,7 +651,7 @@ def _prepare_output_docstrings(output_type, config_class): >>> import tensorflow as tf >>> 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="tf") >>> input_ids = inputs["input_ids"] @@ -669,7 +669,7 @@ def _prepare_output_docstrings(output_type, config_class): >>> import tensorflow as tf >>> tokenizer = {tokenizer_class}.from_pretrained('{checkpoint}') - >>> model = {model_class}.from_pretrained('{checkpoint}', return_dict=True)) + >>> model = {model_class}.from_pretrained('{checkpoint}', return_dict=True) >>> question, text = "Who was Jim Henson?", "Jim Henson was a nice puppet" >>> input_dict = tokenizer(question, text, return_tensors='tf') @@ -688,7 +688,7 @@ def _prepare_output_docstrings(output_type, config_class): >>> import tensorflow as tf >>> 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="tf") >>> inputs["labels"] = tf.reshape(tf.constant(1), (-1, 1)) # Batch size 1 @@ -705,7 +705,7 @@ def _prepare_output_docstrings(output_type, config_class): >>> import tensorflow as tf >>> 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("The capital of France is {mask}.", return_tensors="tf") >>> inputs["labels"] = tokenizer("The capital of France is Paris.", return_tensors="tf")["input_ids"] @@ -722,7 +722,7 @@ def _prepare_output_docstrings(output_type, config_class): >>> import tensorflow as tf >>> 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="tf") >>> outputs = model(inputs) @@ -737,7 +737,7 @@ def _prepare_output_docstrings(output_type, config_class): >>> import tensorflow as tf >>> tokenizer = {tokenizer_class}.from_pretrained('{checkpoint}') - >>> model = {model_class}.from_pretrained('{checkpoint}', return_dict=True)) + >>> model = {model_class}.from_pretrained('{checkpoint}', return_dict=True) >>> prompt = "In Italy, pizza served in formal settings, such as at a restaurant, is presented unsliced." >>> choice0 = "It is eaten with a fork and a knife." @@ -758,7 +758,7 @@ def _prepare_output_docstrings(output_type, config_class): >>> import tensorflow as tf >>> 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="tf") >>> outputs = model(inputs) From 80b0c27f82cb9ac781782d9188a93958c8651b62 Mon Sep 17 00:00:00 2001 From: mymusise Date: Thu, 12 Nov 2020 23:43:06 +0800 Subject: [PATCH 2/2] fix example bug Signed-off-by: mymusise --- docs/source/main_classes/trainer.rst | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/source/main_classes/trainer.rst b/docs/source/main_classes/trainer.rst index a181c4f5524e..12fff9d518a0 100644 --- a/docs/source/main_classes/trainer.rst +++ b/docs/source/main_classes/trainer.rst @@ -35,7 +35,7 @@ Here is an example of how to customize :class:`~transformers.Trainer` using a cu class MyTrainer(Trainer): def compute_loss(self, model, inputs): labels = inputs.pop("labels") - outputs = models(**inputs) + outputs = model(**inputs) logits = outputs[0] return my_custom_loss(logits, labels)