From 4d1a2096ad9b8c6a2118f96a24f7664d249073c3 Mon Sep 17 00:00:00 2001
From: dohe0342 <kimdohe1070@gmail.com>
Date: Sat, 10 Dec 2022 14:04:56 +0900
Subject: [PATCH] from local

---
 .../.train.py.swp                             | Bin 102400 -> 102400 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 c25d286a4ff5404da63bd681bd0ba96daf0ea7a6..25125747759dd9c4b12dc462af1173684e7eb3df 100644
GIT binary patch
delta 185
zcmZozz}B#UO*F|M%+puFQqO<^2m}}ye1)eZ*KHKF7G#Xt94NT?9B(Ky1H(-gh-B1e
z!3XS&scSeG7&?Hs0f=3I*cpf|fmj=eKe00~JO$#JK->ewjzFvg#QZ>ff{lTp7KoF8
zSQ3aYvNAB71mXf94gz9bApQ-s>J|{+++6!GhI#YzC)b%d%ad&V^7B%46gKaC;l3OI
DJlrSe

delta 179
zcmZozz}B#UO*F|M%+puFQqO<^2m}}y9E7JNhi??M7G#Xv94NT?9B(=^1H*F`h-Bnu
z!3XS&sna+Z82W%X2#5oL*d2&1ftUe^KeID1EC=FQK<p3120$ze#Mjsu7`lPD0EpFr
x_&zHG!*w981L6cAwgzHOAbtU~3S`~Y&9x6>m?xLi2yedk<P!7deJ`Ar0{|XwCF1}9

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 baba0ae06..199cd8a50 100755
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
@@ -720,6 +720,7 @@ def compute_loss(
     sp: spm.SentencePieceProcessor,
     batch: dict,
     is_training: bool,
+    wb=None, 
 ) -> Tuple[Tensor, MetricsTracker]:
     """
     Compute transducer loss given the model and its inputs.