diff --git a/openwakeword/train.py b/openwakeword/train.py index 411fc43..4df0798 100755 --- a/openwakeword/train.py +++ b/openwakeword/train.py @@ -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