10 Commits

Author SHA1 Message Date
Ross Wightman
1420c118df Missed comitting outstanding changes to default_cfg keys and test exclusions for swin v2 2022-02-23 19:50:26 -08:00
Ross Wightman
c6e4b7895a Swin V2 CR impl refactor.
* reformat and change some naming so closer to existing timm vision transformers
* remove typing that wasn't adding clarity (or causing torchscript issues)
* support non-square windows
* auto window size adjust from image size
* post-norm + main-branch no
2022-02-23 17:28:52 -08:00
Christoph Reich
67d140446b Fix bug in classification head 2022-02-20 22:28:05 +01:00
Christoph Reich
29add820ac Refactor (back to relative imports) 2022-02-20 00:46:48 +01:00
Christoph Reich
74a04e0016 Add parameter to change normalization type 2022-02-20 00:46:00 +01:00
Christoph Reich
2a4f6c13dd Create model functions 2022-02-20 00:40:22 +01:00
Christoph Reich
87b4d7a29a Add get and reset classifier method 2022-02-19 22:47:02 +01:00
Christoph Reich
ff5f6bcd6c Check input resolution 2022-02-19 22:42:02 +01:00
Christoph Reich
81bf0b4033 Change parameter names to match Swin V1 2022-02-19 22:37:22 +01:00
Christoph Reich
f227b88831 Add initials (CR) to model and file 2022-02-19 22:14:38 +01:00