Update optim.py (#1366)

k2-fsa · Nov 3, 2023 · 231bbcd · 231bbcd
1 parent c3bbb32
commit 231bbcd
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/egs/librispeech/ASR/zipformer/optim.py b/egs/librispeech/ASR/zipformer/optim.py
@@ -466,6 +466,8 @@ def _get_clipping_scale(
                 quartiles.append(sorted_norms[index].item())
 
             median = quartiles[2]
+            if median - median != 0:
+                raise RuntimeError("Too many grads were not finite")
             threshold = clipping_scale * median
             if step in irregular_estimate_steps:
                 # use larger thresholds on first few steps of estimating threshold,