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)]
|
self._init_state: List[torch.Tensor] = [torch.empty(0)]
|
||||||
|
|
||||||
def forward(
|
def forward(
|
||||||
self, x: torch.Tensor, x_lens: torch.Tensor, warmup: float = 1.0, layer_output = False,
|
self, x: torch.Tensor, x_lens: torch.Tensor, warmup: float = 1.0, get_layer_output = False,
|
||||||
) -> Tuple[torch.Tensor, torch.Tensor]:
|
) -> Tuple[torch.Tensor, torch.Tensor, Optional[List[torch.Tensor]]]:
|
||||||
"""
|
"""
|
||||||
Args:
|
Args:
|
||||||
x:
|
x:
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user