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.

Sign up for GitHub

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

Jump to bottom

Diff tool #168

Merged
merged 15 commits into from
Aug 26, 2024
Merged

Diff tool #168

merged 15 commits into from
Aug 26, 2024

Conversation

justinchuby
Copy link
Owner

@justinchuby justinchuby commented Aug 26, 2024 •
edited
Loading

Create a diff tool that can diff two ONNX models, or an ONNX model and an ExportedProgram.

Example usage:

import torch_onnx.tools.diff_model

torch_onnx.tools.diff_model.diff_exported_program(
    "senet154.onnx",  # The onnx model to diff
    onnx_program.exported_program,  # The reference exported program
    ["relu_2", "getitem_9"],  # Intermediate values to diff
    data,  # input data
    keep_original_outputs=True,  # Whether to retain original model outputs
)

@codecov Codecov
Copy link

codecov bot commented Aug 26, 2024 •
edited
Loading

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 59.04%. Comparing base (c4607b7) to head (d0e2db0).
Report is 1 commits behind head on main.

Additional details and impacted files 
@@           Coverage Diff           @@
##             main     #168   +/-   ##
=======================================
  Coverage   59.04%   59.04%           
=======================================
  Files          22       22           
  Lines        3003     3003           
  Branches      636      636           
=======================================
  Hits         1773     1773           
  Misses       1037     1037           
  Partials      193      193

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

github-advanced-security[bot]
github-advanced-security bot found potential problems Aug 26, 2024
View reviewed changes
src/torch_onnx/tools/diff_model.py Fixed Show fixed Hide fixed
src/torch_onnx/tools/diff_model.py Fixed Show fixed Hide fixed
src/torch_onnx/tools/diff_model.py Fixed Show fixed Hide fixed
src/torch_onnx/tools/diff_model.py Fixed Show fixed Hide fixed
src/torch_onnx/tools/diff_model.py Fixed Show fixed Hide fixed
src/torch_onnx/tools/diff_model.py Fixed Show fixed Hide fixed
src/torch_onnx/tools/diff_model.py Fixed Show fixed Hide fixed
@justinchuby justinchuby merged commit 03aabd5 into main Aug 26, 2024
7 checks passed
@justinchuby justinchuby deleted the diff-tool branch August 26, 2024 23:24
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers
No reviews
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
1 participant
@justinchuby