From fe35141e7ea04754f321de3180c6d54beb720de8 Mon Sep 17 00:00:00 2001 From: jinzr Date: Fri, 17 Nov 2023 17:10:04 +0800 Subject: [PATCH] Update train.py --- egs/multi_zh_en/ASR/zipformer/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/multi_zh_en/ASR/zipformer/train.py b/egs/multi_zh_en/ASR/zipformer/train.py index e1bf65da9..b9e0a9a5a 100755 --- a/egs/multi_zh_en/ASR/zipformer/train.py +++ b/egs/multi_zh_en/ASR/zipformer/train.py @@ -1172,7 +1172,7 @@ def run(rank, world_size, args): if params.print_diagnostics: opts = diagnostics.TensorDiagnosticOptions( - 2**22 + 512 ) # allow 4 megabytes per sub-module diagnostic = diagnostics.attach_diagnostics(model, opts)