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