added log to search for

mosaicml · Apr 2, 2024 · 6a7746e · 6a7746e
1 parent be48d2d
commit 6a7746e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/llmfoundry/data/packing.py b/llmfoundry/data/packing.py
@@ -451,6 +451,7 @@ def profile(raw_batch_size: int) -> Tuple[Optional[float], Optional[float]]:
         waste_percent = 100 * packer.waste
         return padding_percent, waste_percent
 
+    log.info('Profiling packing ratios')
     with logging_redirect_tqdm(loggers=[log]):
         for packing_ratio, raw_batch_size in (pbar := tqdm(
                 zip(packing_ratios, raw_batch_sizes),