from local

This commit is contained in:
dohe0342 2023-01-09 20:35:04 +09:00
parent 1a3323e574
commit 186d756804
3 changed files with 1 additions and 0 deletions

View File

@ -197,6 +197,7 @@ class Conformer(EncoderInterface):
x = x.permute(1, 0, 2) # (T, N, C) ->(N, T, C)
layer_outputs = [x.permute(1, 0, 2) for x in layer_outputs]
layer_outputs.append(pos_emb)
if get_layer_output:
return x, lengths, layer_outputs