Skip to content
This repository has been archived by the owner on Mar 12, 2024. It is now read-only.

update position encoding to make detr torch fx traceable #531

Open
wants to merge 2 commits into
base: main
Choose a base branch
from
Open

update position encoding to make detr torch fx traceable #531

wants to merge 2 commits into from

Conversation

inisis
Copy link

@inisis inisis commented Aug 31, 2022

update position encoding to make detr torch fx traceable

@facebook-github-bot facebook-github-bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Aug 31, 2022
@inisis
Copy link
Author

inisis commented Aug 31, 2022

without modification, using torch fx to trace detr will fail.
image
this is a problem related to torch arange fucntion, and a workaround is to warp it,
image
and replace torch.arange with torch_arange.
image
after this, detr can be traced by torch fx

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Reviewers
No reviews
Assignees
No one assigned
Labels
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
2 participants
@inisis @facebook-github-bot