from local

This commit is contained in:
dohe0342 2023-01-09 19:29:16 +09:00
parent dc477d9711
commit 5f6b801d41
3 changed files with 1 additions and 8 deletions

View File

@ -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: