logging

mlcommons · Oct 3, 2023 · 8594d24 · 8594d24
1 parent cbee8f0
commit 8594d24
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/algorithmic_efficiency/data_utils.py b/algorithmic_efficiency/data_utils.py
@@ -81,6 +81,7 @@ def _prepare(x):
     logging.info(new_batch['inputs'])
     logging.info(len(new_batch['inputs']))
     logging.info(np.array(new_batch['inputs'][0]).shape)
+    logging.info(np.array(new_batch['inputs'][1]).shape)
     # logging.info("Padded weights: ")
   return new_batch