akadmin/openwakeword
1
0
Fork 0
mirror of https://github.com/dscripka/openWakeWord.git synced 2026-08-27 18:17:20 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

set opset version for onnx model export [skip ci]

Browse source
This commit is contained in:
dscripka 2023-10-25 12:29:50 -04:00
parent ef118bd271
commit e9fd49d533
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
openwakeword/train.py
View file

@ -309,7 +309,7 @@ class Model(nn.Module):
# Save ONNX model
logging.info(f"####\nSaving ONNX mode as '{os.path.join(output_dir, model_name + '.onnx')}'")
model_to_save = copy.deepcopy(model)
torch.onnx.export(model_to_save.to("cpu"), torch.rand(self.input_shape)[None, ], os.path.join(output_dir, model_name + ".onnx"))
torch.onnx.export(model_to_save.to("cpu"), torch.rand(self.input_shape)[None, ], os.path.join(output_dir, model_name + ".onnx"), opset_version=13)
return None