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
8b8bf8250b
commit
c681ce6c65
Binary file not shown.
@ -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
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user