16 Commits (9a86b900fa6ea62d39ecd7c7a95e9523fe4cf264)

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