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 = []
|
||||
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
|
||||
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user