Update train.py (#1590)

k2-fsa · Apr 11, 2024 · ed6bc20 · ed6bc20
1 parent ba5b2e8
commit ed6bc20
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/egs/librispeech/ASR/zipformer/train.py b/egs/librispeech/ASR/zipformer/train.py
@@ -966,7 +966,10 @@ def save_bad_model(suffix: str = ""):
             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