From 2fe4af8c99be7ff296ec53f083451f0115934329 Mon Sep 17 00:00:00 2001
From: Daniel Povey <dpovey@gmail.com>
Date: Fri, 17 Jun 2022 12:35:55 +0800
Subject: [PATCH] Prevent crash due to error in C_diff

---
 egs/librispeech/ASR/pruned_transducer_stateless7/optim.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/egs/librispeech/ASR/pruned_transducer_stateless7/optim.py b/egs/librispeech/ASR/pruned_transducer_stateless7/optim.py
index c80276710..1a87c92d9 100644
--- a/egs/librispeech/ASR/pruned_transducer_stateless7/optim.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless7/optim.py
@@ -709,7 +709,9 @@ class NeutralGradient(Optimizer):
             C_diff = C_check - C
             # Roundoff can cause significant differences, so use a fairly large
             # threshold of 0.001.  We may increase this later or even remove the check.
-            assert C_diff.abs().mean() < 0.01 * C.diag().mean()
+            if not C_diff.abs().mean() < 0.01 * C.diag().mean():
+                print("Warning: large C_diff: {C_diff.abs().mean()}, C diag mean: {C.diag().mean()}")
+
 
         return P