undo debug

mlcommons · Apr 24, 2024 · 7ed8e6d · 7ed8e6d
1 parent 7477578
commit 7ed8e6d
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/prize_qualification_baselines/external_tuning/jax_nadamw_full_budget.py b/prize_qualification_baselines/external_tuning/jax_nadamw_full_budget.py
@@ -346,6 +346,4 @@ def data_selection(workload: spec.Workload,
   del global_step
   del rng
   batch = next(input_queue)
-  print('BATCH STATS')
-  print(sum(batch['weights']))
   return batch
diff --git a/scoring/run_workloads.py b/scoring/run_workloads.py
@@ -142,7 +142,7 @@ def main(_):
     workload_metadata = json.load(f)
 
   # workloads = [w for w in workload_metadata.keys()]
-  workloads = ['wmt']
+  workloads = []
 
   # Read held-out workloads
   if FLAGS.held_out_workloads_config_path: