From c12b23a73b7e4332ecad9d80e931458d60e7d5d2 Mon Sep 17 00:00:00 2001
From: dohe0342 <kimdohe1070@gmail.com>
Date: Sat, 10 Dec 2022 15:06:17 +0900
Subject: [PATCH] from local

---
 .../.train.py.swp                             | Bin 110592 -> 110592 bytes
 .../train.py                                  |   1 +
 2 files changed, 1 insertion(+)

diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train.py.swp
index 8236023747cd63ca42debe18d1aa7e23527fc098..0fe1c626a2fd732b6211d9831b02eaf887d35130 100644
GIT binary patch
delta 197
zcmZp8z}E19O*F|M%+puFQqO<^2m}}yVicz&f8HosEyx(Pd7@zPIbKg@28OdN5Xq>`
zf)CgkQx|YBFw_8X2@qQVu{jXy0<k<0KVxTLxCz9)K->Vtra&wP#LPgvhmC=u5QxKo
zm=}nTvNAC21>zJSb^~HXAbt<D>KqWC-CX-HfO+%tC)b%pQ&N-jQ&O!IlJfI&6l@jT
P5_5`EbvEyO;eH$dm?SI}

delta 202
zcmZp8z}E19O*F|M%+puFQqO<^2m}}ygcPSF3vLvx7G#XvJW;Ut9B(=^1H*F`h-Bnu
z!3XS&sna+Z82W%X2#5oL*d2&1ftUe^KeID1EC=FQK<p3120$ze#Mjsu7`lPD0EpFr
z_&zHG!*w981L6cAwgzHOAbtU~3S`}*&9x5$n57hK6_OH5k`+LJ0Ys#wCYFF1o6kPE
L&b)cw3#a1%gx)GG

diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
index 096995340..6cc2856c5 100755
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
@@ -726,6 +726,7 @@ def compute_loss(
     sp: spm.SentencePieceProcessor,
     batch: dict,
     is_training: bool,
+    decode: bool = False,
 ) -> Tuple[Tensor, MetricsTracker]:
     """
     Compute transducer loss given the model and its inputs.