mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-09 10:02:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

Small fix (#1686)

Browse Source
This commit is contained in:
Yifan Yang 2024-07-11 14:45:35 +08:00 committed by GitHub
parent 785f3f0bcf
commit d65187ec52
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
egs/librispeech/ASR/zipformer/scaling.py
View File

@ -636,8 +636,9 @@ class ChunkCausalDepthwiseConv1d(torch.nn.Module):
) )
def forward(self, x: Tensor, chunk_size: int = -1) -> Tensor: def forward(self, x: Tensor, chunk_size: int = -1) -> Tensor:
""" """Forward function.
Forward function. Args:
Args:
x: a Tensor of shape (batch_size, channels, seq_len) x: a Tensor of shape (batch_size, channels, seq_len)
chunk_size: the chunk size, in frames; does not have to divide seq_len exactly. chunk_size: the chunk size, in frames; does not have to divide seq_len exactly.
""" """

4
egs/librispeech/ASR/zipformer/train.py
View File

@ -406,7 +406,7 @@ def get_parser():
"--context-size", "--context-size",
type=int, type=int,
default=2, default=2,
help="The context size in the decoder. 1 means bigram; " "2 means tri-gram", help="The context size in the decoder. 1 means bigram; 2 means tri-gram",
) )
parser.add_argument( parser.add_argument(
@ -429,7 +429,7 @@ def get_parser():
"--am-scale", "--am-scale",
type=float, type=float,
default=0.0, default=0.0,
help="The scale to smooth the loss with am (output of encoder network)" "part.", help="The scale to smooth the loss with am (output of encoder network) part.",
) )
parser.add_argument( parser.add_argument(