Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Check if onnx model already converted to fp16 #293

Merged
merged 1 commit into from
Jun 14, 2024

Conversation

xiaowuhu
Copy link
Collaborator

@xiaowuhu xiaowuhu commented Jun 13, 2024

Check if Onnx Model already be converted to fp16 before, by checking:

  • if graph input, output has fp16 type
  • if graph value_info has fp16 type
  • if graph initializer has fp16 type
  • if graph node exist cast to fp16 node.

If yes, raise exception. But use can use check_fp16_ready=False to skip this checking and force to convert the model to fp16 (maybe again).

@xiaowuhu xiaowuhu requested a review from fatcat-z June 13, 2024 07:13
@xiaowuhu xiaowuhu merged commit 209a47e into master Jun 14, 2024
9 checks passed
@xiaowuhu xiaowuhu deleted the xiaowu/checkIfFp16Ready branch June 14, 2024 00:56
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants