from local

This commit is contained in:
dohe0342 2023-02-14 17:56:43 +09:00
parent bce8655058
commit 91c5db76e1
2 changed files with 4 additions and 0 deletions

View File

@ -447,6 +447,10 @@ class ConformerEncoder(nn.Module):
if i in self.aux_layers:
outputs.append(output)
if i+1 in [3,6,9,12,15] and condition_layer is not None:
ctc_out = self.ctc_output(output, log_prob=False)
output = output + condition_layer
#output = self.combiner(outputs)
return output, layer_outputs