Add Swin Transformer models from https://github.com/microsoft/Swin-Transformer

Author: rwightman

tests/test_models.py

@@ -15,7 +15,7 @@
     torch._C._jit_set_profiling_mode(False)
 
 # transformer models don't support many of the spatial / feature based model functionalities
-NON_STD_FILTERS = ['vit_*', 'tnt_*', 'pit_*']
+NON_STD_FILTERS = ['vit_*', 'tnt_*', 'pit_*', 'swin_*']
 NUM_NON_STD = len(NON_STD_FILTERS)
 
 # exclude models that cause specific test failures

timm/models/__init__.py

@@ -26,6 +26,7 @@
 from .selecsls import *
 from .senet import *
 from .sknet import *
+from .swin_transformer import *
 from .tnt import *
 from .tresnet import *
 from .vgg import *

timm/models/pit.py

@@ -31,7 +31,7 @@ def _cfg(url='', **kwargs):
     return {
         'url': url,
         'num_classes': 1000, 'input_size': (3, 224, 224), 'pool_size': None,
-        'crop_pct': .9, 'interpolation': 'bicubic',
+        'crop_pct': .9, 'interpolation': 'bicubic', 'fixed_input_size': True,
         'mean': IMAGENET_DEFAULT_MEAN, 'std': IMAGENET_DEFAULT_STD,
         'first_conv': 'patch_embed.conv', 'classifier': 'head',
         **kwargs