from local

This commit is contained in:
dohe0342 2023-01-09 13:02:49 +09:00
parent 4f0df83455
commit 8f1fdada00
3 changed files with 3 additions and 3 deletions

View File

@ -701,10 +701,10 @@ class ConformerEncoder(nn.Module):
src_key_padding_mask=src_key_padding_mask, src_key_padding_mask=src_key_padding_mask,
warmup=warmup, warmup=warmup,
) )
if i in self.aux_layers: #if i in self.aux_layers:
outputs.append(output) # outputs.append(output)
output = self.combiner(outputs) #output = self.combiner(outputs)
return output return output