update
Browse files
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=
|
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",
|