From 3c8967a695c97aae367b3dfba1ade783e2a95d61 Mon Sep 17 00:00:00 2001 From: Juhan Bae Date: Thu, 21 Sep 2023 11:54:20 -0400 Subject: [PATCH] minor --- algorithmic_efficiency/workloads/criteo1tb/workload.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/algorithmic_efficiency/workloads/criteo1tb/workload.py b/algorithmic_efficiency/workloads/criteo1tb/workload.py index 6a351e4f1..c47d3c77e 100644 --- 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}