Update scripts/train/train.py

mosaicml · May 24, 2024 · 3d0e578 · 3d0e578
1 parent f298c7b
commit 3d0e578
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/train/train.py b/scripts/train/train.py
@@ -531,7 +531,7 @@ def main(cfg: DictConfig) -> Trainer:
     if train_cfg.log_config:
         log.info('Logging config')
         log_config(logged_cfg)
-    log_dataset_uri(train_cfg)
+    log_dataset_uri(logged_cfg)
     torch.cuda.empty_cache()
     gc.collect()