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
dc477d9711
commit
5f6b801d41
Binary file not shown.
Binary file not shown.
@ -242,14 +242,7 @@ class Interformer(nn.Module):
|
|||||||
self,
|
self,
|
||||||
x: torch.Tensor,
|
x: torch.Tensor,
|
||||||
x_lens: torch.Tensor,
|
x_lens: torch.Tensor,
|
||||||
y: k2.RaggedTensor,
|
):
|
||||||
prune_range: int = 5,
|
|
||||||
am_scale: float = 0.0,
|
|
||||||
lm_scale: float = 0.0,
|
|
||||||
warmup: float = 1.0,
|
|
||||||
reduction: str = "sum",
|
|
||||||
delay_penalty: float = 0.0,
|
|
||||||
) -> Tuple[torch.Tensor, torch.Tensor]:
|
|
||||||
"""
|
"""
|
||||||
Args:
|
Args:
|
||||||
x:
|
x:
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user