From e80c46aa68d941195082c733a562ed6ed4a22e66 Mon Sep 17 00:00:00 2001
From: Vasilis Vryniotis <vvryniotis@fb.com>
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 <vvryniotis@fb.com>
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
zcmV<C02%+Q2df9LKmmUODlb191V2B3I3&O5ftx?NMg>1WUID$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
zcmV<C02%+Q2df9LKmmVxFL^#v&fGo%-h@7^EQdZsA5c9$EfGG@?aMtB{6N1r?s-4J
zuSmU6hfcf+;IqDfO4z%cT2Q|p89%-Rv@5?Sv2MN_d~ZLv_8>pyca6V1D5gD(aX>vR
z_YpqWCn&#+QpY~<hlxHg6Qn*|dN{tIyh^>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
zcmV<C02%+Q2df9LKmmVj91FkVp1!`yrT)Gai%Gw5yyiYCf}}nkJT||hq;$Sv5Q)An
zDmp%bJEOi}Phr0lQ$@b2@R`2xmk2-En7KZr)oVV5UAR7eSgt<b*o{6l@%%pguK_=F
zy_r7lqTRk2w4^<gg*ZNHRR29dBuqZ=r$W74ZI?cH?DW08FpW{a4wha%c>m5k(D{Hr
z>k%Zs9oDVB`w@@6fLhPJ+p&DVu&5kApth~P_I7ALW<G_!6a=O|a-@wvj$PEfaLL}i
gh+gx)lCPUSP)i30b`a5#lMn*X1a=V7k(1;C$K{xF;s5{u

delta 230
zcmV<C02%+Q2df9LKmmU(lT^P6jY7YmiB`Ys09C)~udO~(FN;3A>y5wO7wf%%a%sCG
zLMlE8V<Eqdp}xM}MWsFu%Rs*)`!7FGP$EC(+l)Pg4AVUX@hd(vLY6+o2LL}C4A?!6
z2%Wx({{6nt@wz_SZNa^sMdLmgM5Dcrc1%9NW~@Hq`l!7`B|}lal$$?302lW>9i}2b
z2M+PRa9@VLS??D<bQjq^@ezf;d*LTPmq-1+m?B$0m7IdV0M9Nw;dPTg*5I<f5M|`N
g80Xi%Dj}ObP)i30-f(8=lMn*X1m19F>67FF$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)}