Add all transforms to train script (#1300)

mosaicml · Jun 23, 2024 · 8b5a1bb · 8b5a1bb
1 parent 2196d07
commit 8b5a1bb
Showing 1 changed file with 1 addition and 2 deletions.
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,
@@ -197,7 +196,7 @@ def main(cfg: DictConfig) -> Trainer:
         cfg,
         TrainConfig,
         TRAIN_CONFIG_KEYS,
-        transforms=[update_batch_size_info],
+        transforms='all',
     )
 
     # Set logging level