padding

mlcommons · Sep 29, 2023 · b7b94a2 · b7b94a2
1 parent 6fdc23a
commit b7b94a2
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/algorithmic_efficiency/data_utils.py b/algorithmic_efficiency/data_utils.py
@@ -38,6 +38,7 @@ def shard_and_maybe_pad_np(
   remainder_size = current_batch_size % local_device_count
   # if pad_to_global_batch_size:
   if remainder_size != 0:
+    logging.info("PADDDINGGGGGGG")
     if global_batch_size is not None:
       pad_size = global_batch_size - current_batch_size
     else:
@@ -62,6 +63,7 @@ def _prepare(x):
     # Pad if remainder_size != 0 (should only be possible during evaluation).
     # if pad_to_global_batch_size:
     if remainder_size != 0: 
+      logging.info("PADDDINGGGGGG in _prepare")
       x = pad(x, pad_size, padding_value=padding_value)
 
     # Reshape (global_batch_size, ...) to