minor

mlcommons · Sep 21, 2023 · 3c8967a · 3c8967a
1 parent 7ec6cf7
commit 3c8967a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/algorithmic_efficiency/workloads/criteo1tb/workload.py b/algorithmic_efficiency/workloads/criteo1tb/workload.py
@@ -149,5 +149,5 @@ def _eval_model_on_split(self,
     if USE_PYTORCH_DDP:
       dist.all_reduce(loss)
 
-    mean_loss = loss.item() / num_examples
+    mean_loss = loss.item()
     return {'loss': mean_loss}