from local

This commit is contained in:
dohe0342 2023-01-29 19:33:39 +09:00
parent 83629cb7d7
commit 1c28ecd7d9
2 changed files with 1 additions and 1 deletions

View File

@ -696,7 +696,7 @@ class ConformerEncoder(nn.Module):
residual = None residual = None
for i, mod in enumerate(self.layers): for i, mod in enumerate(self.layers):
if i == 2: if i == 0:
residual = output residual = output
elif i in [2,5,8]: elif i in [2,5,8]:
residual = output residual = output