loss weights fixed

* I have to add weight support for all regression function in the library

loss weights fixed
* I have to add weight support for all regression function in the library
b885ba8a · Pavol Mulinka · cd5d5b41 · b885ba8a
隐藏空白更改
内联并排

Showing with 0 addition and 1 deletion

pytorch_widedeep/training/trainer.py pytorch_widedeep/training/trainer.py +0 -1

未找到文件。
--- a/pytorch_widedeep/training/trainer.py
+++ b/pytorch_widedeep/training/trainer.py
@@ -1220,7 +1220,6 @@ class Trainer:
                        UserWarning,
                    )
                    loss = self.loss_fn(y_pred, y)
-
            else:
                loss = self.loss_fn(y_pred, y)