From 1d8283617aca2ca1bb0263839784e8f756a8a477 Mon Sep 17 00:00:00 2001 From: Juhan Bae Date: Tue, 19 Sep 2023 14:28:16 -0400 Subject: [PATCH] minor --- algorithmic_efficiency/workloads/criteo1tb/workload.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/algorithmic_efficiency/workloads/criteo1tb/workload.py b/algorithmic_efficiency/workloads/criteo1tb/workload.py index 65354f822..c36835704 100644 --- a/algorithmic_efficiency/workloads/criteo1tb/workload.py +++ b/algorithmic_efficiency/workloads/criteo1tb/workload.py @@ -135,8 +135,7 @@ def _eval_model_on_split(self, repeat_final_dataset=False) loss = 0.0 size = 0 - for i in range(num_batches): - eval_batch = next(self._eval_iters[split]) + for eval_batch in self._eval_iters[split]: # if i == (num_batches - 1): # print(eval_batch.get('weights')) loss += self._eval_batch(params, eval_batch)