Minor Swin Transformer fixes

hubconf.py

@@ -67,6 +67,7 @@
     shufflenet_v2_x2_0,
 )
 from torchvision.models.squeezenet import squeezenet1_0, squeezenet1_1
+from torchvision.models.swin_transformer import swin_t, swin_s, swin_b
 from torchvision.models.vgg import vgg11, vgg13, vgg16, vgg19, vgg11_bn, vgg13_bn, vgg16_bn, vgg19_bn
 from torchvision.models.vision_transformer import (
     vit_b_16,

test/test_models.py

@@ -327,6 +327,8 @@ def _check_input_backprop(model, inputs):
     "regnet_x_16gf",
     "regnet_x_32gf",
     "swin_t",
+    "swin_s",
+    "swin_b",
 ]
 for m in slow_models:
     _model_params[m] = {"input_shape": (1, 3, 64, 64)}