Skip to content

Refactor swin transfomer so later we can reuse component for 3d version #6088

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.

Sign up for GitHub

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

Merged
merged 19 commits into from
May 26, 2022
Merged

Refactor swin transfomer so later we can reuse component for 3d version #6088

Changes from 14 commits
Commits
Show all changes
19 commits
Select commit Hold shift + click to select a range
fb1a749
Use List[int] instead of int for window_size and shift_size
YosuaMichael May 25, 2022
c3902ae
Make PatchMerging and SwinTransformerBlock able to handle 2d and 3d c…
YosuaMichael May 25, 2022
13d4d85
Separate patch embedding from SwinTransformer and enable to get model…
YosuaMichael May 25, 2022
a3d1192
Dont use if before padding so it is fx friendly
YosuaMichael May 25, 2022
6a57c15
Merge branch 'models/refactor-swin-transformer' into models/refactor-…
YosuaMichael May 25, 2022
ba5f8f9
Put the handling on window_size edge cases on separate function and w…
YosuaMichael May 25, 2022
ed14bd7
Update the weight url to the converted weight with new structure
YosuaMichael May 25, 2022
548109a
Update the accuracy of swin_transformer
YosuaMichael May 25, 2022
5d09c35
Merge pull request #2 from YosuaMichael/models/refactor-swin-transfor…
YosuaMichael May 25, 2022
2d767f8
Change assert to Exception and nit
YosuaMichael May 25, 2022
65c8439
Make num_classes optional
YosuaMichael May 25, 2022
2c0dead
Merge branch 'main' into models/refactor-swin-transformer
YosuaMichael May 25, 2022
f0f872f
Add typing output for _fix_window_and_shift_size function
YosuaMichael May 26, 2022
e2317f4
init head to None to make it jit scriptable
YosuaMichael May 26, 2022
77ce2b9
Revert the change to make num_classes optional
YosuaMichael May 26, 2022
f04801f
Revert unneccesarry changes that might be risky
YosuaMichael May 26, 2022
480f762
Merge branch 'main' into models/refactor-swin-transformer
YosuaMichael May 26, 2022
8cf5e39
Remove self.head declaration
YosuaMichael May 26, 2022
c260edb
Merge branch 'models/refactor-swin-transformer' of github.com:YosuaMi…
YosuaMichael May 26, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Loading