SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fix doc bug (#8500)

Browse Source 
* fix doc bug

Signed-off-by: mymusise <mymusise1@gmail.com>

* fix example bug

Signed-off-by: mymusise <mymusise1@gmail.com>
This commit is contained in:
Chengxi Guo
2020-11-13 00:47:23 +08:00
committed by GitHub
parent 924c624a46
commit d65e0bfea3
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
docs/source/main_classes/trainer.rst
View File

@@ -35,7 +35,7 @@ Here is an example of how to customize :class:`~transformers.Trainer` using a cu
class MyTrainer(Trainer): class MyTrainer(Trainer):
def compute_loss(self, model, inputs): def compute_loss(self, model, inputs):
labels = inputs.pop("labels") labels = inputs.pop("labels")
outputs = models(**inputs) outputs = model(**inputs)
logits = outputs[0] logits = outputs[0]
return my_custom_loss(logits, labels) return my_custom_loss(logits, labels)