count #batches instead of #users

PreferredAI · Oct 23, 2023 · ebce517 · ebce517
1 parent 7c7ff21
commit ebce517
Showing 1 changed file with 2 additions and 2 deletions.
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(reduction="sum")
+        criteria = nn.BCELoss()
 
         loop = trange(self.num_epochs, disable=not self.verbose)
         for _ in loop:
@@ -241,7 +241,7 @@ def _fit_pt(self):
                 loss.backward()
                 optimizer.step()
 
-                count += len(batch_users)
+                count += 1
                 sum_loss += loss.data.item()
 
                 if batch_id % 10 == 0: