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,
|
||||
x: 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:
|
||||
x:
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user