From fdfaf2d6b620ecd167a48fe7b8b22e1ad1921f6b Mon Sep 17 00:00:00 2001 From: yh_cc Date: Sun, 14 Apr 2019 22:38:15 +0800 Subject: [PATCH] =?UTF-8?q?=E9=98=B2=E6=AD=A2=E5=A4=9A=E5=8D=A1=E7=9A=84?= =?UTF-8?q?=E6=83=85=E5=86=B5=E5=AF=BC=E8=87=B4=E6=97=A0=E6=B3=95=E6=AD=A3?= =?UTF-8?q?=E7=A1=AE=E8=AE=A1=E7=AE=97loss=E2=80=9C?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- fastNLP/core/trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/fastNLP/core/trainer.py b/fastNLP/core/trainer.py index 2a8d85da..b45dd148 100644 --- a/fastNLP/core/trainer.py +++ b/fastNLP/core/trainer.py @@ -265,7 +265,7 @@ class Trainer(object): # edit prediction self.callback_manager.on_loss_begin(batch_y, prediction) - loss = self._compute_loss(prediction, batch_y) + loss = self._compute_loss(prediction, batch_y).mean() avg_loss += loss.item() loss = loss/self.update_every