from local

This commit is contained in:
dohe0342 2023-01-29 19:34:50 +09:00
parent 1c28ecd7d9
commit 4aa3796210
2 changed files with 1 additions and 2 deletions

View File

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