mirrors/icefall
Archived
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix causal

Browse Source
This commit is contained in:
pkufool 2022-07-25 14:56:53 +08:00
parent d773b29db2
commit 245515fbfe
1 changed files with 9 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
egs/librispeech/ASR/pruned_transducer_stateless5/conformer.py
View File

@ -113,17 +113,18 @@ class Conformer(EncoderInterface):
self.num_left_chunks = num_left_chunks self.num_left_chunks = num_left_chunks
encoder_layer = ConformerEncoderLayer( encoder_layer = ConformerEncoderLayer(
d_model, d_model=d_model,
nhead, nhead=nhead,
dim_feedforward, dim_feedforward=dim_feedforward,
dropout, dropout=dropout,
layer_dropout, layer_dropout=layer_dropout,
cnn_module_kernel, cnn_module_kernel=cnn_module_kernel,
causal=causal,
) )
# aux_layers from 1/3 # aux_layers from 1/3
self.encoder = ConformerEncoder( self.encoder = ConformerEncoder(
encoder_layer, encoder_layer=encoder_layer,
num_encoder_layers, num_encoder_layers=num_encoder_layers,
aux_layers=list( aux_layers=list(
range( range(
num_encoder_layers // 3, num_encoder_layers // 3,