from local

2025-12-11 06:55:27 +00:00 · 2022-12-10 15:06:17 +09:00 · 2022-12-10 15:06:17 +09:00 · c12b23a73b
commit c12b23a73b
parent e9d5b46d53
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
@ -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.