From e9668a5cfd484cf6b1cd9387cb7f1cbf4c218dae Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Sun, 18 Jun 2023 03:36:00 +0800 Subject: [PATCH] Fix break in fix_diagnostics mode --- egs/libriheavy/LM/zipformer1/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/libriheavy/LM/zipformer1/train.py b/egs/libriheavy/LM/zipformer1/train.py index e3964f6bf..3fda65eb4 100755 --- a/egs/libriheavy/LM/zipformer1/train.py +++ b/egs/libriheavy/LM/zipformer1/train.py @@ -794,7 +794,7 @@ def train_one_epoch( display_and_save_batch(batch, params=params) raise - if params.print_diagnostics and batch_idx == 5: + if params.print_diagnostics and _batch_idx == 5: return if (