fix BCELoss, reduction='sum'

PreferredAI · Oct 23, 2023 · 7c7ff21 · 7c7ff21
1 parent d300be2
commit 7c7ff21
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cornac/models/ncf/recom_ncf_base.py b/cornac/models/ncf/recom_ncf_base.py
@@ -218,7 +218,7 @@ def _fit_pt(self):
             lr=self.lr,
             weight_decay=self.reg,
         )
-        criteria = nn.BCELoss()
+        criteria = nn.BCELoss(reduction="sum")
 
         loop = trange(self.num_epochs, disable=not self.verbose)
         for _ in loop: