removed logging exception

mosaicml · Jun 19, 2024 · 2a773b6 · 2a773b6
1 parent 4b1fecb
commit 2a773b6
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/scripts/train/train.py b/scripts/train/train.py
@@ -394,9 +394,7 @@ def main(cfg: DictConfig) -> Trainer:
             train_cfg.device_train_batch_size,
         )
     except BaseContextualError as e:
-        if mosaicml_logger is not None:
-            e.location = TrainDataLoaderLocation
-            mosaicml_logger.log_exception(e)
+        e.location = TrainDataLoaderLocation
         raise e
 
     if mosaicml_logger is not None:
@@ -427,9 +425,7 @@ def main(cfg: DictConfig) -> Trainer:
             if eval_gauntlet_callback is not None:
                 callbacks.append(eval_gauntlet_callback)
         except BaseContextualError as e:
-            if mosaicml_logger is not None:
-                e.location = EvalDataLoaderLocation
-                mosaicml_logger.log_exception(e)
+            e.location = EvalDataLoaderLocation
             raise e
 
     if mosaicml_logger is not None: