diff --git a/neural_lam/train_model.py b/neural_lam/train_model.py
index 9d1d5039..74146c89 100644
--- a/neural_lam/train_model.py
+++ b/neural_lam/train_model.py
@@ -23,7 +23,7 @@
 }
 
 
-@logger.catch(reraise=True)
+@logger.catch
 def main(input_args=None):
     """Main function for training and evaluating models."""
     parser = ArgumentParser(