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
for i,j in enumerate(indexes):
mod = self.layers[j]
for i in indexes:
mod = self.layers[i]
output, attn_scores = mod(
output,
pos_emb,