pc

mosaicml · Jun 22, 2024 · c16e94d · c16e94d
1 parent 4df1ebd
commit c16e94d
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/scripts/train/train.py b/scripts/train/train.py
@@ -50,7 +50,6 @@
     make_dataclass_and_log_config,
     pop_config,
     process_init_device,
-    update_batch_size_info,
 )
 from llmfoundry.utils.exceptions import (
     BaseContextualError,