From 5a8cabd429478c595ff46ada4d6194882bc8b687 Mon Sep 17 00:00:00 2001
From: Daniel Povey <dpovey@gmail.com>
Date: Sun, 18 Jun 2023 03:28:57 +0800
Subject: [PATCH] Fix max_eig  arg to TensorDiagnosticsOptions

---
 egs/libriheavy/LM/zipformer1/train.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/egs/libriheavy/LM/zipformer1/train.py b/egs/libriheavy/LM/zipformer1/train.py
index b9478134b..e3964f6bf 100755
--- a/egs/libriheavy/LM/zipformer1/train.py
+++ b/egs/libriheavy/LM/zipformer1/train.py
@@ -980,8 +980,8 @@ def run(rank, world_size, args):
 
     if params.print_diagnostics:
         opts = diagnostics.TensorDiagnosticOptions(
-            2 ** 22
-        )  # allow 4 megabytes per sub-module
+            512,
+        )
         diagnostic = diagnostics.attach_diagnostics(model, opts)
 
     if params.inf_check: