Skip to content

Commit

Permalink
ok
Browse files Browse the repository at this point in the history
  • Loading branch information
johndpope committed Oct 5, 2024
1 parent 77514d9 commit db89f72
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion onnxconv.py
Original file line number Diff line number Diff line change
Expand Up @@ -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=21,
opset_version=20,
do_constant_folding=True,
input_names=['x_current', 'x_reference'],
output_names=['f_r', 't_r', 't_c'],
Expand Down Expand Up @@ -214,6 +214,8 @@ def export_to_onnx(model, x_current, x_reference, file_name):
print(f"\nConverted and validated model saved to {web_compatible_file}")
print("This model should now be compatible with WONNX")

print(f"nnx info {web_compatible_file} > output.txt 2>&1")

except Exception as e:
print(f"Error during ONNX export and validation: {str(e)}")
import traceback
Expand Down

0 comments on commit db89f72

Please sign in to comment.