from local

This commit is contained in:
dohe0342 2023-01-30 14:49:52 +09:00
parent c56229baf0
commit da07c74fb6
2 changed files with 1 additions and 1 deletions

View File

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