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

[TorchFX] Support output port id #3109

Open
1 task
daniil-lyakhov opened this issue Nov 25, 2024 · 0 comments
Open
1 task

[TorchFX] Support output port id #3109

daniil-lyakhov opened this issue Nov 25, 2024 · 0 comments
Labels
enhancement New feature or request

Comments

@daniil-lyakhov
Copy link
Collaborator

🚀 Feature request

Some TorchFX operations are returning a tuple instead of a torch.Tesnor. In case an algorithm requests a statistic from such node nncf will raise an error like tuple has not is_empty method. This affect the yolon11 model, current workaround is the ignored scope with the type of operation which return a tuple:

nncf.IgnoredScope(types=["__getitem__"])

The task is to:

  • Introduce FXTargetPoint with port_id attr instead of input_port_id
  • Support FXTargetPoint in each TorchFX transformation

Feature Use Case

No response

Are you going to submit a PR?

  • Yes I'd like to help by submitting a PR!
@daniil-lyakhov daniil-lyakhov added the enhancement New feature or request label Nov 25, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

No branches or pull requests

1 participant