From e80c46aa68d941195082c733a562ed6ed4a22e66 Mon Sep 17 00:00:00 2001 From: Vasilis Vryniotis Date: Thu, 19 May 2022 16:25:57 +0100 Subject: [PATCH 1/2] Add swin on hubconfig. --- hubconf.py | 1 + 1 file changed, 1 insertion(+) 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, From c4893b8ffadfa02d98fd660c6a2d3e65a34571ff Mon Sep 17 00:00:00 2001 From: Vasilis Vryniotis Date: Thu, 19 May 2022 16:26:26 +0100 Subject: [PATCH 2/2] Add swin b/s in the `slow_models` list. --- test/expect/ModelTester.test_swin_b_expect.pkl | Bin 939 -> 939 bytes test/expect/ModelTester.test_swin_s_expect.pkl | Bin 939 -> 939 bytes test/test_models.py | 2 ++ 3 files changed, 2 insertions(+) diff --git a/test/expect/ModelTester.test_swin_b_expect.pkl b/test/expect/ModelTester.test_swin_b_expect.pkl index d807ca3ed1588c47c932891b26b99835c3f86d94..2ae40af400fa9a201530fbb9760d67397be0812a 100644 GIT binary patch delta 230 zcmV1WUID$L^^!ht@y@>d`Jg}e zy8FIW)jhwkfxx_M>KeX$sfoVQ0TI91ZM{C$DJi~wC9FUCyiz{wmr%czNV&agi2}ZQ zD$Tz3S{uLjUN=7g77sp|YqLHGn%BLlmRi1+&<#J*{U<**mts-AA`LCSIgZi33wcdH z6JQ*_{H}>VYY^!^bgnkO?gatA9vK9`d>co<&oocJfTvPAXo9gn#LAIB9#l)dtB}(^ gn#Nf_k|dD6P)i30v5x|SlMn*X1hJ0-gOlU}#~y`h-2eap delta 230 zcmVpyca6V1D5gD(aX>vR z_YpqWCn&#+QpY~R%Va+lQei(37+6t0?4_nYn~W7cIFDOB z#uLrIO~lkbq&G@FHy#APZ23VxhDB??LQ0apC|6OxAcWyP6tf6F{BJ2gFr0$ESo^EK giNBma@{zc{P)i30yYI)elMn*X1iSCYw3Fll$F>A$bN~PV diff --git a/test/expect/ModelTester.test_swin_s_expect.pkl b/test/expect/ModelTester.test_swin_s_expect.pkl index 2624dad4178f435ef0cb5a2a306da88b66846cbf..b8b31bab413cdbd970a04a6ae1b8203ac12561dd 100644 GIT binary patch delta 230 zcmVm5k(D{Hr z>k%Zs9oDVB`w@@6fLhPJ+p&DVu&5kApth~P_I7ALWy5wO7wf%%a%sCG zLMlE8V9i}2b z2M+PRa9@VLS??D67FF$Bf5p$^ZZW 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)}