from local

This commit is contained in:
dohe0342 2023-01-21 15:30:57 +09:00
parent 8b8bf8250b
commit c681ce6c65
2 changed files with 4 additions and 0 deletions

View File

@ -328,6 +328,8 @@ class ConformerEncoder(nn.Module):
""" """
output = src output = src
layer_outputs = []
for i, mod in enumerate(self.layers): for i, mod in enumerate(self.layers):
output = mod( output = mod(
output, output,
@ -337,6 +339,8 @@ class ConformerEncoder(nn.Module):
warmup=warmup, warmup=warmup,
) )
layer_outputs.append(output)
return output return output