from local

This commit is contained in:
dohe0342 2023-01-09 19:41:49 +09:00
parent 9e21e1f0e1
commit 8d43329787
2 changed files with 11 additions and 1 deletions

View File

@ -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)]