Skip to content

Commit e65372e

Browse files
authored
Minor Swin Transformer fixes (#6054)
* Add swin on hubconfig. * Add swin b/s in the `slow_models` list.
1 parent 9d9cfab commit e65372e

File tree

4 files changed

+3
-0
lines changed

4 files changed

+3
-0
lines changed

hubconf.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -67,6 +67,7 @@
6767
shufflenet_v2_x2_0,
6868
)
6969
from torchvision.models.squeezenet import squeezenet1_0, squeezenet1_1
70+
from torchvision.models.swin_transformer import swin_t, swin_s, swin_b
7071
from torchvision.models.vgg import vgg11, vgg13, vgg16, vgg19, vgg11_bn, vgg13_bn, vgg16_bn, vgg19_bn
7172
from torchvision.models.vision_transformer import (
7273
vit_b_16,
0 Bytes
Binary file not shown.
0 Bytes
Binary file not shown.

test/test_models.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -327,6 +327,8 @@ def _check_input_backprop(model, inputs):
327327
"regnet_x_16gf",
328328
"regnet_x_32gf",
329329
"swin_t",
330+
"swin_s",
331+
"swin_b",
330332
]
331333
for m in slow_models:
332334
_model_params[m] = {"input_shape": (1, 3, 64, 64)}

0 commit comments

Comments
 (0)