set breakpoint

mlcommons · Apr 24, 2024 · 5398d30 · 5398d30
1 parent 516fb7c
commit 5398d30
Showing 1 changed file with 3 additions and 1 deletion.
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,8 @@ def data_selection(workload: spec.Workload,
   del global_step
   del rng
   batch = next(input_queue)
+  breakpoint()
   print('BATCH STATS')
-  print(sum(batch['weights']))
+
+  print((batch['weights']).shape)
   return batch