"docs/vscode:/vscode.git/clone" did not exist on "ebc4edfe7a4dfb820bf7f4faa02fb6de479b6e36"
Unverified Commit 6f041fcb authored by mchau's avatar mchau Committed by GitHub
Browse files

fix documentation for CustomTrainer (#25635)

fix doc
parent 8608bf20
...@@ -60,7 +60,7 @@ from transformers import Trainer ...@@ -60,7 +60,7 @@ from transformers import Trainer
class CustomTrainer(Trainer): class CustomTrainer(Trainer):
def compute_loss(self, model, inputs, return_outputs=False): def compute_loss(self, model, inputs, return_outputs=False):
labels = inputs.get("labels") labels = inputs.pop("labels")
# forward pass # forward pass
outputs = model(**inputs) outputs = model(**inputs)
logits = outputs.get("logits") logits = outputs.get("logits")
......
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