d4c0588012
Remove persistent buffers from Swin-V2. Change SwinV2Cr cos attn + tau/logit_scale to match official, add ckpt convert, init_value zeros resid LN weight by default
2022-05-13 10:50:59 -07:00
c0211b0bf7
Swin-V2 test fixes, typo
2022-05-12 22:31:55 -07:00
9a86b900fa
Official SwinV2 models
2022-05-12 15:05:10 -07:00
b7cb8d0337
Add Swin-V2 Small-NS weights (83.5 @ 224). Add layer scale like 'init_values' via post-norm LN weight scaling
2022-04-26 17:32:49 -07:00
c5a8e929fb
Add initial swinv2 tiny / small weights
2022-04-03 15:22:55 -07:00
024fc4d9ab
version 0.6.1 for master
2022-03-21 22:03:13 -07:00
e1e037ba52
Fix bad tuple typing fix that was on XLA branch bust missed on master merge
2022-03-21 22:00:33 -07:00
fe457c1996
Update SwinTransformerV2Cr post-merge, update with grad checkpointing / grad matcher
...
* weight compat break, activate norm3 for final block of final stage (equivalent to pre-head norm, but while still in BLC shape)
* remove fold/unfold for TPU compat, add commented out roll code for TPU
* add option for end of stage norm in all stages
* allow weight_init to be selected between pytorch default inits and xavier / moco style vit variant
2022-03-21 14:50:28 -07:00
1420c118df
Missed comitting outstanding changes to default_cfg keys and test exclusions for swin v2
2022-02-23 19:50:26 -08:00
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
67d140446b
Fix bug in classification head
2022-02-20 22:28:05 +01:00
29add820ac
Refactor (back to relative imports)
2022-02-20 00:46:48 +01:00
74a04e0016
Add parameter to change normalization type
2022-02-20 00:46:00 +01:00
2a4f6c13dd
Create model functions
2022-02-20 00:40:22 +01:00
87b4d7a29a
Add get and reset classifier method
2022-02-19 22:47:02 +01:00
ff5f6bcd6c
Check input resolution
2022-02-19 22:42:02 +01:00
81bf0b4033
Change parameter names to match Swin V1
2022-02-19 22:37:22 +01:00
f227b88831
Add initials (CR) to model and file
2022-02-19 22:14:38 +01:00
Ross Wightman
Christoph Reich