fix causal
This commit is contained in:
parent
d773b29db2
commit
245515fbfe
@ -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,
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user