From 2a773b6536e1be1e7eee07ff783193008d727e74 Mon Sep 17 00:00:00 2001 From: jjanezhang Date: Wed, 19 Jun 2024 16:58:36 -0700 Subject: [PATCH] removed logging exception --- scripts/train/train.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/scripts/train/train.py b/scripts/train/train.py index f2a70b526d..3799a3eb7f 100644 --- 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: