from local

2022-12-10 14:04:56 +09:00 · 2022-12-10 14:04:56 +09:00 · 4d1a2096ad
commit 4d1a2096ad
parent d195287c51
2 changed files with 1 additions and 0 deletions
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train.py.swp
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train.py.swp
--- 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.