Merge pull request #265 from eltociear/patch-2

chore: update encoder.py
This commit is contained in:
Xiang Lyu
2024-08-10 21:38:53 +08:00
committed by GitHub

View File

@@ -297,7 +297,7 @@ class BaseEncoder(torch.nn.Module):
rate.
3. Currently, nn.Sequential is used to stack all the convolution
layers in subsampling, we need to rewrite it to make it work
with cache, which is not prefered.
with cache, which is not preferred.
Args:
xs (torch.Tensor): (1, max_len, dim)
chunk_size (int): decoding chunk size