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
f007665de4
commit
1a3323e574
Binary file not shown.
Binary file not shown.
@ -247,7 +247,6 @@ class Interformer(nn.Module):
|
|||||||
with torch.no_grad():
|
with torch.no_grad():
|
||||||
encoder_out, x_lens, pt_layer_outputs = self.pt_encoder(x,
|
encoder_out, x_lens, pt_layer_outputs = self.pt_encoder(x,
|
||||||
x_lens,
|
x_lens,
|
||||||
warmup=warmup,
|
|
||||||
get_layer_output=True
|
get_layer_output=True
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user