mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
from local
This commit is contained in:
parent
e8c4d4fe6c
commit
21ef5e476c
Binary file not shown.
Binary file not shown.
@ -512,7 +512,10 @@ class Tempformer(EncoderInterface):
|
|||||||
):
|
):
|
||||||
layer_outputs = []
|
layer_outputs = []
|
||||||
for enum, encoder in enumerate(self.encoder_layers):
|
for enum, encoder in enumerate(self.encoder_layers):
|
||||||
layer_outputs.append(encoder(x[enum]))
|
x_ = x[enum].permute(1, 0, 2)
|
||||||
|
encoder_output = encoder(x_)
|
||||||
|
encoder_output = encoder_output.permute(1, 0, 2)
|
||||||
|
layer_outputs.append(encoder(x_))
|
||||||
|
|
||||||
return layer_outputs
|
return layer_outputs
|
||||||
|
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user