mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-12-11 06:55:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

from local

Browse Source
This commit is contained in:
dohe0342 2023-01-09 19:27:00 +09:00
parent 7a2448d887
commit 06dca57f4b
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
egs/librispeech/ASR/incremental_transf/.conformer.py.swp
View File

Binary file not shown.

4
egs/librispeech/ASR/incremental_transf/conformer.py
View File

@ -136,7 +136,7 @@ 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 self, x: torch.Tensor, x_lens: torch.Tensor, warmup: float = 1.0, layer_output = False,
) -> Tuple[torch.Tensor, torch.Tensor]: ) -> Tuple[torch.Tensor, torch.Tensor]:
""" """
Args: Args:
@ -198,7 +198,7 @@ class Conformer(EncoderInterface):
x = x.permute(1, 0, 2) # (T, N, C) ->(N, T, C) x = x.permute(1, 0, 2) # (T, N, C) ->(N, T, C)
layer_outputs = [x.permute(1, 0, 2) for x in layer_outputs] layer_outputs = [x.permute(1, 0, 2) for x in layer_outputs]
return x, lengths, layer_outputs return x, lengths
@torch.jit.export @torch.jit.export
def get_init_state( def get_init_state(