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

ONNX Floor and Resize Operations Not Supported #500

Open
Yoshi234 opened this issue Mar 28, 2024 · 0 comments
Open

ONNX Floor and Resize Operations Not Supported #500

Yoshi234 opened this issue Mar 28, 2024 · 0 comments

Comments

@Yoshi234
Copy link

Hi,

I'm trying to use CrypTen's crypten.nn.from_pytorch() method to load YOLO5 from ultralytics. However, I have been running into some issues with ONNX functions Floor and Resize operations not being supported. I have seen other issues that discuss using the crypten.nn.module.py file to add additional functionality, but I'm unsure of how to add floor with the functions provided by the default cryptensor methods. I was considering adding methods to the mpc tensor structures themselves but I'm not exactly sure how to do this either, or if this is even the best approach to take.

Do you have any recommendations for me, and do you know how I should go about implementing a new cryptensor method if I wanted to do that (not just a module built on top of existing cryptensor methods)?

Thanks for the help!

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

No branches or pull requests

1 participant