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
06dca57f4b
commit
cdd062a4fe
Binary file not shown.
@ -136,8 +136,8 @@ class Conformer(EncoderInterface):
|
||||
self._init_state: List[torch.Tensor] = [torch.empty(0)]
|
||||
|
||||
def forward(
|
||||
self, x: torch.Tensor, x_lens: torch.Tensor, warmup: float = 1.0, layer_output = False,
|
||||
) -> Tuple[torch.Tensor, torch.Tensor]:
|
||||
self, x: torch.Tensor, x_lens: torch.Tensor, warmup: float = 1.0, get_layer_output = False,
|
||||
) -> Tuple[torch.Tensor, torch.Tensor, Optional[List[torch.Tensor]]]:
|
||||
"""
|
||||
Args:
|
||||
x:
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user