mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-08-09 01:52:41 +00:00
delete duplicate line for encoder initial state (#765)
This commit is contained in:
parent
ad475ec10d
commit
6d659f423d
@ -152,7 +152,6 @@ def export_encoder_model_jit_trace(
|
|||||||
|
|
||||||
x = torch.zeros(1, T, 80, dtype=torch.float32)
|
x = torch.zeros(1, T, 80, dtype=torch.float32)
|
||||||
states = encoder_model.init_states()
|
states = encoder_model.init_states()
|
||||||
states = encoder_model.init_states()
|
|
||||||
|
|
||||||
traced_model = torch.jit.trace(encoder_model, (x, states))
|
traced_model = torch.jit.trace(encoder_model, (x, states))
|
||||||
traced_model.save(encoder_filename)
|
traced_model.save(encoder_filename)
|
||||||
|
Loading…
x
Reference in New Issue
Block a user