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:39:39 +09:00
parent 8df7e1fda4
commit 9e21e1f0e1
2 changed files with 15 additions and 21 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.

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

@ -480,6 +480,7 @@ class Tempformer(EncoderInterface):
self.short_chunk_size = short_chunk_size self.short_chunk_size = short_chunk_size
self.num_left_chunks = num_left_chunks self.num_left_chunks = num_left_chunks
def build_conformer(d_model, nhead, dim_feedforward, dropout, layer_dropout, cnn_module_kernel, causal):
encoder_layer = ConformerEncoderLayer( encoder_layer = ConformerEncoderLayer(
d_model=d_model, d_model=d_model,
nhead=nhead, nhead=nhead,
@ -489,18 +490,11 @@ class Tempformer(EncoderInterface):
cnn_module_kernel=cnn_module_kernel, cnn_module_kernel=cnn_module_kernel,
causal=causal, causal=causal,
) )
# aux_layers from 1/3 return encoder_layer
self.encoder = ConformerEncoder(
encoder_layer=encoder_layer,
num_layers=num_encoder_layers, self.encoder_layers = nn.ModuleList(
aux_layers=list(
range(
num_encoder_layers // 3,
num_encoder_layers - 1,
aux_layer_period,
)
),
)
self._init_state: List[torch.Tensor] = [torch.empty(0)] self._init_state: List[torch.Tensor] = [torch.empty(0)]
def forward( def forward(