fix causal

This commit is contained in:
pkufool 2022-07-25 14:56:53 +08:00
parent d773b29db2
commit 245515fbfe

View File

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