15 Commits (d07d0151738417dc754e6620656e9e9a9621aae8)

Author	SHA1	Message	Date
Ross Wightman	b7cb8d0337	Add Swin-V2 Small-NS weights (83.5 @ 224). Add layer scale like 'init_values' via post-norm LN weight scaling	2 years ago
Ross Wightman	c5a8e929fb	Add initial swinv2 tiny / small weights	2 years ago
Ross Wightman	024fc4d9ab	version 0.6.1 for master	2 years ago
Ross Wightman	e1e037ba52	Fix bad tuple typing fix that was on XLA branch bust missed on master merge	2 years ago
Ross Wightman	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	2 years ago
Ross Wightman	1420c118df	Missed comitting outstanding changes to default_cfg keys and test exclusions for swin v2	2 years ago
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	2 years ago
Christoph Reich	67d140446b	Fix bug in classification head	2 years ago
Christoph Reich	29add820ac	Refactor (back to relative imports)	2 years ago
Christoph Reich	74a04e0016	Add parameter to change normalization type	2 years ago
Christoph Reich	2a4f6c13dd	Create model functions	2 years ago
Christoph Reich	87b4d7a29a	Add get and reset classifier method	2 years ago
Christoph Reich	ff5f6bcd6c	Check input resolution	2 years ago
Christoph Reich	81bf0b4033	Change parameter names to match Swin V1	2 years ago
Christoph Reich	f227b88831	Add initials (CR) to model and file	2 years ago