From 6844b13299d9891cecf809dd243d0e4d5bf87933 Mon Sep 17 00:00:00 2001 From: Juhan Bae Date: Tue, 19 Sep 2023 14:21:23 -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 7c73e1863..2597e8904 100644 --- a/algorithmic_efficiency/workloads/criteo1tb/workload.py +++ b/algorithmic_efficiency/workloads/criteo1tb/workload.py @@ -132,7 +132,7 @@ def _eval_model_on_split(self, split=split, data_dir=data_dir, global_batch_size=global_batch_size, - repeat_final_dataset=True) + repeat_final_dataset=False) loss = 0.0 for i in range(num_batches): eval_batch = next(self._eval_iters[split])