from local

This commit is contained in:
dohe0342 2023-01-09 19:24:26 +09:00
parent 3ae7169f5d
commit 1c11c19869
2 changed files with 2 additions and 1 deletions

View File

@ -703,12 +703,13 @@ class ConformerEncoder(nn.Module):
warmup=warmup, warmup=warmup,
) )
layer_outputs.append(output) layer_outputs.append(output)
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, layer_outputs
@torch.jit.export @torch.jit.export
def chunk_forward( def chunk_forward(