clean_up

mlcommons · Mar 12, 2024 · a8108d6 · a8108d6
1 parent 85ae1af
commit a8108d6
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 3 deletions.
diff --git a/algorithmic_efficiency/workloads/criteo1tb/workload.py b/algorithmic_efficiency/workloads/criteo1tb/workload.py
@@ -5,7 +5,6 @@
 from typing import Dict, Iterator, Optional, Tuple
 
 from absl import flags
-from absl import logging
 import torch.distributed as dist
 
 from algorithmic_efficiency import spec
@@ -142,7 +141,6 @@ def _eval_model_on_split(self,
     num_batches = int(math.ceil(num_examples / global_batch_size))
     if split not in self._eval_iters:
       # These iterators will repeat indefinitely.
-      logging.info(f"BUILDING split with {data_dir} ")
       self._eval_iters[split] = self._build_input_queue(
           data_rng=rng,
           split=split,

diff --git a/submission_runner.py b/submission_runner.py
@@ -622,7 +622,6 @@ def score_submission_on_workload(workload: spec.Workload,
 
 
 def main(_):
-  logging.info(f'Data directory is set to {FLAGS.data_dir}')
   if FLAGS.profile:
     profiler = Profiler()
   else: