from local

This commit is contained in:
dohe0342 2023-01-29 19:31:14 +09:00
parent eabb297212
commit 83629cb7d7
2 changed files with 3 additions and 1 deletions

View File

@ -696,7 +696,9 @@ class ConformerEncoder(nn.Module):
residual = None
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
output = mod(
output,