From 7f323f16021ea034fb71e22d80adef2deda45955 Mon Sep 17 00:00:00 2001
From: Pavol Mulinka <36483359+5uperpalo@users.noreply.github.com>
Date: Mon, 6 Dec 2021 12:41:26 +0100
Subject: [PATCH] Update trainer.py

---
 pytorch_widedeep/training/trainer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pytorch_widedeep/training/trainer.py b/pytorch_widedeep/training/trainer.py
index 5afa5e1..4d32c82 100644
--- a/pytorch_widedeep/training/trainer.py
+++ b/pytorch_widedeep/training/trainer.py
@@ -1168,7 +1168,7 @@ class Trainer:
         else:
             loss = self.loss_fn(y_pred, y)
             score = self._get_score(y_pred, y)
-        # raise exception if the loss is exploding with non scaled target values
+        # TODO raise exception if the loss is exploding with non scaled target values
         loss.backward()
         self.optimizer.step()
 
-- 
GitLab