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

update position encoding to make detr torch fx traceable #531

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

update position encoding to make detr torch fx traceable #531

inisis 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