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

flake8 fix

Browse source
This commit is contained in:
dscripka 2023-11-10 07:42:17 -05:00
parent d5d377a623
commit fad5ee7db2
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
openwakeword/train.py
View file

@ -309,7 +309,8 @@ 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"), opset_version=13)
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