From ab91ffff8948c7882d662af6014656cb8d00d137 Mon Sep 17 00:00:00 2001
From: Yifan Yang <64255737+yfyeung@users.noreply.github.com>
Date: Thu, 11 Apr 2024 19:23:09 +0800
Subject: [PATCH] Update train.py

---
 egs/librispeech/ASR/zipformer/train.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/egs/librispeech/ASR/zipformer/train.py b/egs/librispeech/ASR/zipformer/train.py
index 1111d32ab..04caf2fd8 100755
--- a/egs/librispeech/ASR/zipformer/train.py
+++ b/egs/librispeech/ASR/zipformer/train.py
@@ -966,7 +966,10 @@ def train_one_epoch(
             scaler.step(optimizer)
             scaler.update()
             optimizer.zero_grad()
-        except:  # noqa
+        except Exception as e:
+            logging.info(
+                f"Caught exception: {e}."
+            )
             save_bad_model()
             display_and_save_batch(batch, params=params, sp=sp)
             raise