minor

mlcommons · Sep 21, 2023 · da1392f · da1392f
1 parent b6398a9
commit da1392f
Showing 1 changed file with 4 additions and 9 deletions.
diff --git a/algorithmic_efficiency/workloads/criteo1tb/workload.py b/algorithmic_efficiency/workloads/criteo1tb/workload.py
@@ -63,13 +63,13 @@ def num_eval_train_examples(self) -> int:
 
   @property
   def num_validation_examples(self) -> int:
-    # return 89_000_000
-    return 83_296_256
+    return 89_000_000
+    # return 83_296_256
 
   @property
   def num_test_examples(self) -> int:
-    # return 89_274_637
-    return 95_027_200
+    return 89_274_637
+    # return 95_027_200
 
   @property
   def train_mean(self):
@@ -137,14 +137,9 @@ def _eval_model_on_split(self,
           repeat_final_dataset=True)
     loss = 0.0
 
-    # for eval_batch in self._eval_iters[split]:
     for _ in range(num_batches):
       eval_batch = next(self._eval_iters[split])
-      # if i == (num_batches - 1):
-      #   print(eval_batch.get('weights'))
       loss += self._eval_batch(params, eval_batch)
-      # size += eval_batch.get('weights').sum()
-      # print(eval_batch["inputs"].shape[0])
 
     if USE_PYTORCH_DDP:
       dist.all_reduce(loss)