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",
|