mirror of
https://github.com/modelscope/FunASR
synced 2025-09-15 14:48:36 +08:00
update
This commit is contained in:
parent
ce2f775f44
commit
e1549946bc
@ -43,7 +43,7 @@ from funasr.modules.subsampling import TooShortUttError
|
|||||||
from funasr.modules.subsampling import check_short_utt
|
from funasr.modules.subsampling import check_short_utt
|
||||||
from funasr.modules.subsampling import Conv2dSubsamplingPad
|
from funasr.modules.subsampling import Conv2dSubsamplingPad
|
||||||
|
|
||||||
class ConvolutionModule(AbsEncoder):
|
class ConvolutionModule(nn.Module):
|
||||||
"""ConvolutionModule in Conformer model.
|
"""ConvolutionModule in Conformer model.
|
||||||
|
|
||||||
Args:
|
Args:
|
||||||
@ -278,7 +278,7 @@ class EncoderLayer(nn.Module):
|
|||||||
return x, mask
|
return x, mask
|
||||||
|
|
||||||
|
|
||||||
class ConformerEncoder(torch.nn.Module):
|
class ConformerEncoder(AbsEncoder):
|
||||||
"""Conformer encoder module.
|
"""Conformer encoder module.
|
||||||
|
|
||||||
Args:
|
Args:
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user