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
eabb297212
commit
83629cb7d7
Binary file not shown.
@ -696,7 +696,9 @@ class ConformerEncoder(nn.Module):
|
|||||||
residual = None
|
residual = None
|
||||||
|
|
||||||
for i, mod in enumerate(self.layers):
|
for i, mod in enumerate(self.layers):
|
||||||
if i in [2,5,8]:
|
if i == 2:
|
||||||
|
residual = output
|
||||||
|
elif i in [2,5,8]:
|
||||||
residual = output
|
residual = output
|
||||||
output = mod(
|
output = mod(
|
||||||
output,
|
output,
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user