Unverified Commit e65372e1 authored by Vasilis Vryniotis's avatar Vasilis Vryniotis Committed by GitHub
Browse files

Minor Swin Transformer fixes (#6054)

* Add swin on hubconfig.

* Add swin b/s in the `slow_models` list.
parent 9d9cfab2
...@@ -67,6 +67,7 @@ from torchvision.models.shufflenetv2 import ( ...@@ -67,6 +67,7 @@ from torchvision.models.shufflenetv2 import (
shufflenet_v2_x2_0, shufflenet_v2_x2_0,
) )
from torchvision.models.squeezenet import squeezenet1_0, squeezenet1_1 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.vgg import vgg11, vgg13, vgg16, vgg19, vgg11_bn, vgg13_bn, vgg16_bn, vgg19_bn
from torchvision.models.vision_transformer import ( from torchvision.models.vision_transformer import (
vit_b_16, vit_b_16,
......
File suppressed by a .gitattributes entry or the file's encoding is unsupported.
File suppressed by a .gitattributes entry or the file's encoding is unsupported.
...@@ -327,6 +327,8 @@ slow_models = [ ...@@ -327,6 +327,8 @@ slow_models = [
"regnet_x_16gf", "regnet_x_16gf",
"regnet_x_32gf", "regnet_x_32gf",
"swin_t", "swin_t",
"swin_s",
"swin_b",
] ]
for m in slow_models: for m in slow_models:
_model_params[m] = {"input_shape": (1, 3, 64, 64)} _model_params[m] = {"input_shape": (1, 3, 64, 64)}
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment