mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
from local
This commit is contained in:
parent
9e21e1f0e1
commit
8d43329787
Binary file not shown.
@ -493,7 +493,17 @@ class Tempformer(EncoderInterface):
|
||||
return encoder_layer
|
||||
|
||||
|
||||
self.encoder_layers = nn.ModuleList(
|
||||
self.encoder_layers = nn.ModuleList([
|
||||
build_conformer(d_model,
|
||||
nhead,
|
||||
dim_feedforward,
|
||||
dropout,
|
||||
layer_dropout,
|
||||
cnn_module_kernel,
|
||||
causal
|
||||
)
|
||||
for i in range(12)
|
||||
])
|
||||
|
||||
self._init_state: List[torch.Tensor] = [torch.empty(0)]
|
||||
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user