From 75656f1f94a54395e408e2a5cb8ae934e2bd1120 Mon Sep 17 00:00:00 2001 From: John Pope Date: Sun, 6 Oct 2024 09:22:34 +1100 Subject: [PATCH] ok --- onnxconv.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/onnxconv.py b/onnxconv.py index f97ee56..ea61226 100644 --- a/onnxconv.py +++ b/onnxconv.py @@ -150,7 +150,7 @@ def export_to_onnx(model, x_current, x_reference, file_name): (x_current, x_reference), file_name, export_params=True, - opset_version=14, + opset_version=19, do_constant_folding=True, input_names=['x_current', 'x_reference'], output_names=['f_r', 't_r', 't_c'], @@ -161,6 +161,7 @@ def export_to_onnx(model, x_current, x_reference, file_name): 't_r': {0: 'batch_size'}, 't_c': {0: 'batch_size'} }, + export_params=True, verbose=True ) print(f"Model exported successfully to {file_name}")