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,
)
layer_outputs.append(output)
if i in self.aux_layers:
outputs.append(output)
output = self.combiner(outputs)
return output
return output, layer_outputs
@torch.jit.export
def chunk_forward(