Update train.py (#1590)

2025-08-09 10:02:22 +00:00 · 2024-04-11 19:35:25 +08:00 · 2024-04-11 19:35:25 +08:00 · ed6bc200e3
commit ed6bc200e3
parent ba5b2e854b
1 changed files with 4 additions and 1 deletions
--- 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