PR #17182 ViT and Swin symbolic tracing with torch.fx

ViT and Swin symbolic tracing with torch.fx #17182

michaelbenayoun merged 5 commits into huggingface:main from michaelbenayoun:fx_vit_swin

Support tracing for ViT

8641b968

Swin support

bbe205d2

michaelbenayoun force pushed to bbe205d2 3 years ago

Fix copies

4e5ebfff

Fix type annotation issue

6649bc1d

michaelbenayoun changed the title ~~Fx vit swin~~ ViT and Swin symbolic tracing with torch.fx 3 years ago

Removed unused import

9157b19a

michaelbenayoun requested a review from

sgugger 3 years ago

sgugger approved these changes on 2022-05-11

michaelbenayoun merged 8c7481f3 into main 3 years ago

michaelbenayoun deleted the fx_vit_swin branch 3 years ago

Reviewers

sgugger

Assignees

No one assigned

Labels

None yet

Milestone

No milestone