From 2f2934a115062da35c20843b187f2ba61de10678 Mon Sep 17 00:00:00 2001
From: Daniel Povey <dpovey@gmail.com>
Date: Mon, 16 May 2022 19:41:16 +0800
Subject: [PATCH] fix train.py

---
 egs/librispeech/ASR/pruned_transducer_stateless4b/train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/egs/librispeech/ASR/pruned_transducer_stateless4b/train.py b/egs/librispeech/ASR/pruned_transducer_stateless4b/train.py
index f35e4e08f..ba88060e3 100755
--- a/egs/librispeech/ASR/pruned_transducer_stateless4b/train.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless4b/train.py
@@ -703,7 +703,7 @@ def train_one_epoch(
 
         if params.batch_idx_train % 2000 == 0 and params.batch_idx_train > 0:
             mmodel = model.module if hasattr(model, 'module') else model
-            mmodel.encoder.diagonalize()
+            mmodel.diagonalize()
             #optimizer.reset()
 
         params.batch_idx_train += 1