HoneyTian commited on
Commit
12ce37e
·
1 Parent(s): 7b46343
examples/silero_vad_by_webrtcvad/step_5_export_model.py CHANGED
@@ -73,7 +73,7 @@ def main():
73
 
74
  torch.onnx.export(model_export,
75
  args=(inputs, encoder_in_cache, encoder_hidden_cache_list, lstm_hidden_state),
76
- f="silero_vad.onnx",
77
  input_names=["inputs", "encoder_in_cache", "encoder_hidden_cache_list", "lstm_hidden_state"],
78
  output_names=[
79
  "logits", "probs", "lsnr",
 
73
 
74
  torch.onnx.export(model_export,
75
  args=(inputs, encoder_in_cache, encoder_hidden_cache_list, lstm_hidden_state),
76
+ f=output_file.as_posix(),
77
  input_names=["inputs", "encoder_in_cache", "encoder_hidden_cache_list", "lstm_hidden_state"],
78
  output_names=[
79
  "logits", "probs", "lsnr",