diff --git a/hubconf.py b/hubconf.py index a229ab07667..66989fcb281 100644 --- a/hubconf.py +++ b/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, diff --git a/test/expect/ModelTester.test_swin_b_expect.pkl b/test/expect/ModelTester.test_swin_b_expect.pkl index d807ca3ed15..2ae40af400f 100644 Binary files a/test/expect/ModelTester.test_swin_b_expect.pkl and b/test/expect/ModelTester.test_swin_b_expect.pkl differ diff --git a/test/expect/ModelTester.test_swin_s_expect.pkl b/test/expect/ModelTester.test_swin_s_expect.pkl index 2624dad4178..b8b31bab413 100644 Binary files a/test/expect/ModelTester.test_swin_s_expect.pkl and b/test/expect/ModelTester.test_swin_s_expect.pkl differ diff --git a/test/test_models.py b/test/test_models.py index c0afe9f10be..7d29dde0678 100644 --- a/test/test_models.py +++ b/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)}