mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
minor fix of conv-emformer
This commit is contained in:
parent
6c5fd6f648
commit
aff7c4ee3c
@ -552,7 +552,7 @@ class EmformerLayer(nn.Module):
|
|||||||
max_memory_size: int = 0,
|
max_memory_size: int = 0,
|
||||||
tanh_on_mem: bool = False,
|
tanh_on_mem: bool = False,
|
||||||
negative_inf: float = -1e8,
|
negative_inf: float = -1e8,
|
||||||
causal: bool = False,
|
causal: bool = True,
|
||||||
):
|
):
|
||||||
super().__init__()
|
super().__init__()
|
||||||
|
|
||||||
@ -1058,7 +1058,7 @@ class EmformerEncoder(nn.Module):
|
|||||||
max_memory_size: int = 0,
|
max_memory_size: int = 0,
|
||||||
tanh_on_mem: bool = False,
|
tanh_on_mem: bool = False,
|
||||||
negative_inf: float = -1e8,
|
negative_inf: float = -1e8,
|
||||||
causal: bool = False,
|
causal: bool = True,
|
||||||
):
|
):
|
||||||
super().__init__()
|
super().__init__()
|
||||||
|
|
||||||
@ -1088,6 +1088,8 @@ class EmformerEncoder(nn.Module):
|
|||||||
]
|
]
|
||||||
)
|
)
|
||||||
|
|
||||||
|
self.encoder_pos = RelPositionalEncoding(d_model, dropout)
|
||||||
|
|
||||||
self.left_context_length = left_context_length
|
self.left_context_length = left_context_length
|
||||||
self.right_context_length = right_context_length
|
self.right_context_length = right_context_length
|
||||||
self.chunk_length = chunk_length
|
self.chunk_length = chunk_length
|
||||||
@ -1384,7 +1386,7 @@ class Emformer(EncoderInterface):
|
|||||||
max_memory_size: int = 0,
|
max_memory_size: int = 0,
|
||||||
tanh_on_mem: bool = False,
|
tanh_on_mem: bool = False,
|
||||||
negative_inf: float = -1e8,
|
negative_inf: float = -1e8,
|
||||||
causal: bool = False,
|
causal: bool = True,
|
||||||
):
|
):
|
||||||
super().__init__()
|
super().__init__()
|
||||||
|
|
||||||
@ -1555,7 +1557,7 @@ class ConvolutionModule(nn.Module):
|
|||||||
channels: int,
|
channels: int,
|
||||||
kernel_size: int,
|
kernel_size: int,
|
||||||
bias: bool = True,
|
bias: bool = True,
|
||||||
causal: bool = False,
|
causal: bool = True,
|
||||||
) -> None:
|
) -> None:
|
||||||
"""Construct an ConvolutionModule object."""
|
"""Construct an ConvolutionModule object."""
|
||||||
super(ConvolutionModule, self).__init__()
|
super(ConvolutionModule, self).__init__()
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user