From d65e0bfea300bf811b1fef099f01719b1df7f1ce Mon Sep 17 00:00:00 2001 From: Chengxi Guo Date: Fri, 13 Nov 2020 00:47:23 +0800 Subject: [PATCH] Fix doc bug (#8500) * fix doc bug Signed-off-by: mymusise * 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 a181c4f552..12fff9d518 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)