Slight code cleanup/simplification

This commit is contained in:
Daniel Povey 2022-10-06 14:29:51 +08:00
parent 99d17d13cf
commit e1d741a632

View File

@ -355,8 +355,8 @@ class ConformerEncoder(nn.Module):
indexes[j+1] = a indexes[j+1] = a
for i,j in enumerate(indexes): for i in indexes:
mod = self.layers[j] mod = self.layers[i]
output, attn_scores = mod( output, attn_scores = mod(
output, output,
pos_emb, pos_emb,