swin-transformer权重汇总

王正义123

于 2025-03-12 14:23:52 发布

阅读量735

点赞数 10

文章标签： transformer 深度学习人工智能

本文链接：https://blog.youkuaiyun.com/ghuuj90uio/article/details/146204407

版权

default_cfgs = generate_default_cfgs({

'swin_small_patch4_window7_224.ms_in22k_ft_in1k': _cfg(

hf_hub_id='timm/',

url='https://github.com/SwinTransformer/storage/releases/download/v1.0.8/swin_small_patch4_window7_224_22kto1k_finetune.pth', ),

'swin_base_patch4_window7_224.ms_in22k_ft_in1k': _cfg(

hf_hub_id='timm/',

url='https://github.com/SwinTransformer/storage/releases/download/v1.0.0/swin_base_patch4_window7_224_22kto1k.pth',),

'swin_base_patch4_window12_384.ms_in22k_ft_in1k': _cfg(

hf_hub_id='timm/',

url='https://github.com/SwinTransformer/storage/releases/download/v1.0.0/swin_base_patch4_window12_384_22kto1k.pth',

input_size=(3, 384, 384), pool_size=(12, 12), crop_pct=1.0),

'swin_large_patch4_window7_224.ms_in22k_ft_in1k': _cfg(

hf_hub_id='timm/',

url='https://github.com/SwinTransformer/storage/releases/download/v1.0.0/swin_large_patch4_window7_224_22kto1k.pth',),

'swin_large_patch4_window12_384.ms_in22k_ft_in1k': _cfg(

hf_hub_id='timm/',

url='https://github.com/SwinTransformer/storage/releases/download/v1.0.0/swin_large_patch4_window12_384_22kto1k.pth',

input_size=(3, 384, 384), pool_size=(12, 12), crop_pct=1.0),

'swin_tiny_patch4_window7_224.ms_in1k': _cfg(

hf_hub_id='timm/',

url='https://github.com/SwinTransformer/storage/releases/download/v1.0.0/swin_tiny_patch4_window7_224.pth',),

'swin_small_patch4_window7_224.ms_in1k': _cfg(

hf_hub_id='timm/',

url='https://github.com/SwinTransformer/storage/releases/download/v1.0.0/swin_small_patch4_window7_224.pth',),

'swin_base_patch4_window7_224.ms_in1k': _cfg(

hf_hub_id='timm/',

url='https://github.com/SwinTransformer/storage/releases/download/v1.0.0/swin_base_patch4_window7_224.pth',),

'swin_base_patch4_window12_384.ms_in1k': _cfg(

hf_hub_id='timm/',

url='https://github.com/SwinTransformer/storage/releases/download/v1.0.0/swin_base_patch4_window12_384.pth',

input_size=(3, 384, 384), pool_size=(12, 12), crop_pct=1.0),

# tiny 22k pretrain is worse than 1k, so moved after (untagged priority is based on order)

'swin_tiny_patch4_window7_224.ms_in22k_ft_in1k': _cfg(

hf_hub_id='timm/',

url='https://github.com/SwinTransformer/storage/releases/download/v1.0.8/swin_tiny_patch4_window7_224_22kto1k_finetune.pth',),

'swin_tiny_patch4_window7_224.ms_in22k': _cfg(

hf_hub_id='timm/',

url='https://github.com/SwinTransformer/storage/releases/download/v1.0.8/swin_tiny_patch4_window7_224_22k.pth',

num_classes=21841),

'swin_small_patch4_window7_224.ms_in22k': _cfg(

hf_hub_id='timm/',

url='https://github.com/SwinTransformer/storage/releases/download/v1.0.8/swin_small_patch4_window7_224_22k.pth',

num_classes=21841),

'swin_base_patch4_window7_224.ms_in22k': _cfg(

hf_hub_id='timm/',

url='https://github.com/SwinTransformer/storage/releases/download/v1.0.0/swin_base_patch4_window7_224_22k.pth',

num_classes=21841),

'swin_base_patch4_window12_384.ms_in22k': _cfg(

hf_hub_id='timm/',

url='https://github.com/SwinTransformer/storage/releases/download/v1.0.0/swin_base_patch4_window12_384_22k.pth',

input_size=(3, 384, 384), pool_size=(12, 12), crop_pct=1.0, num_classes=21841),

'swin_large_patch4_window7_224.ms_in22k': _cfg(

hf_hub_id='timm/',

url='https://github.com/SwinTransformer/storage/releases/download/v1.0.0/swin_large_patch4_window7_224_22k.pth',

num_classes=21841),

'swin_large_patch4_window12_384.ms_in22k': _cfg(

hf_hub_id='timm/',

url='https://github.com/SwinTransformer/storage/releases/download/v1.0.0/swin_large_patch4_window12_384_22k.pth',

input_size=(3, 384, 384), pool_size=(12, 12), crop_pct=1.0, num_classes=21841),

'swin_s3_tiny_224.ms_in1k': _cfg(

hf_hub_id='timm/',

url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/s3_t-1d53f6a8.pth'),

'swin_s3_small_224.ms_in1k': _cfg(

hf_hub_id='timm/',

url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/s3_s-3bb4c69d.pth'),

'swin_s3_base_224.ms_in1k': _cfg(

hf_hub_id='timm/',

url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/s3_b-a1e95db4.pth'),

})