From 81c71aceb8853f591bb069f8880c3697eead06ca Mon Sep 17 00:00:00 2001 From: Yige Xu Date: Sun, 17 Nov 2019 01:27:54 +0800 Subject: [PATCH] =?UTF-8?q?[bugfix]=E4=BF=AE=E5=A4=8D=E4=BA=86trainer?= =?UTF-8?q?=E5=9C=A8update=5Fevery=E5=A4=A7=E4=BA=8E1=E6=97=B6loss?= =?UTF-8?q?=E7=9A=84=E6=98=BE=E7=A4=BA=E9=94=99=E8=AF=AF?= 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 a39362e2..721478f7 100644 --- a/fastNLP/core/trainer.py +++ b/fastNLP/core/trainer.py @@ -666,8 +666,8 @@ class Trainer(object): # edit prediction self.callback_manager.on_loss_begin(batch_y, prediction) loss = self._compute_loss(prediction, batch_y).mean() - avg_loss += loss.item() loss = loss / self.update_every + avg_loss += loss.item() # Is loss NaN or inf? requires_grad = False self.callback_manager.on_backward_begin(loss)