From 6f041fcbb853adc6c37da85515384ed9a9c5b181 Mon Sep 17 00:00:00 2001
From: mchau <minhtriet09@gmail.com>
Date: Mon, 21 Aug 2023 22:23:17 +0700
Subject: [PATCH] fix documentation for CustomTrainer (#25635)

fix doc
---
 docs/source/en/main_classes/trainer.md | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/docs/source/en/main_classes/trainer.md b/docs/source/en/main_classes/trainer.md
index ad3ea57f13..4a767ee076 100644
--- a/docs/source/en/main_classes/trainer.md
+++ b/docs/source/en/main_classes/trainer.md
@@ -60,7 +60,7 @@ from transformers import Trainer
 
 class CustomTrainer(Trainer):
     def compute_loss(self, model, inputs, return_outputs=False):
-        labels = inputs.get("labels")
+        labels = inputs.pop("labels")
         # forward pass
         outputs = model(**inputs)
         logits = outputs.get("logits")