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

Add perm steerer support #1

Closed

morrismvw wants to merge 3 commits into DavideCatto:main from machineswithvision:add-perm-steerer-support

morrismvw commented Oct 3, 2024 •

edited

Loading

Added ONNX-compatible permutation steerer implementation:

Revised export to rely on scripting instead of tracing
Creating matching-only neural network and related wrapper model for scripting. No need to script the rest of the model being traceable
Setting BatchNorm1d affine flag to True. Workaround for PyTorch bug

See qualitative results:

morrismvw added 3 commits

October 3, 2024 15:09


          Add exportable XFeat with permutation steerers implementation

52615c3


          Update export script

816d5f3


          Update README.md

9f67232

machineswithvision closed this by deleting the head repository

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet