mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
from local
This commit is contained in:
parent
e9d5b46d53
commit
c12b23a73b
Binary file not shown.
@ -726,6 +726,7 @@ def compute_loss(
|
|||||||
sp: spm.SentencePieceProcessor,
|
sp: spm.SentencePieceProcessor,
|
||||||
batch: dict,
|
batch: dict,
|
||||||
is_training: bool,
|
is_training: bool,
|
||||||
|
decode: bool = False,
|
||||||
) -> Tuple[Tensor, MetricsTracker]:
|
) -> Tuple[Tensor, MetricsTracker]:
|
||||||
"""
|
"""
|
||||||
Compute transducer loss given the model and its inputs.
|
Compute transducer loss given the model and its inputs.
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user