Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
vision
Commits
372f4fae
Unverified
Commit
372f4fae
authored
Jan 12, 2023
by
Aditya Oke
Committed by
GitHub
Jan 12, 2023
Browse files
Expose maxvit and swin_v2 models in hubconf.py (#7078)
Co-authored-by:
Nicolas Hug
<
contact@nicolas-hug.com
>
parent
a679c1b5
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
1 deletion
+2
-1
hubconf.py
hubconf.py
+2
-1
No files found.
hubconf.py
View file @
372f4fae
...
@@ -20,6 +20,7 @@ from torchvision.models.efficientnet import (
...
@@ -20,6 +20,7 @@ from torchvision.models.efficientnet import (
)
)
from
torchvision.models.googlenet
import
googlenet
from
torchvision.models.googlenet
import
googlenet
from
torchvision.models.inception
import
inception_v3
from
torchvision.models.inception
import
inception_v3
from
torchvision.models.maxvit
import
maxvit_t
from
torchvision.models.mnasnet
import
mnasnet0_5
,
mnasnet0_75
,
mnasnet1_0
,
mnasnet1_3
from
torchvision.models.mnasnet
import
mnasnet0_5
,
mnasnet0_75
,
mnasnet1_0
,
mnasnet1_3
from
torchvision.models.mobilenetv2
import
mobilenet_v2
from
torchvision.models.mobilenetv2
import
mobilenet_v2
from
torchvision.models.mobilenetv3
import
mobilenet_v3_large
,
mobilenet_v3_small
from
torchvision.models.mobilenetv3
import
mobilenet_v3_large
,
mobilenet_v3_small
...
@@ -68,6 +69,6 @@ from torchvision.models.shufflenetv2 import (
...
@@ -68,6 +69,6 @@ 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_b
,
swin_s
,
swin_t
from
torchvision.models.swin_transformer
import
swin_b
,
swin_s
,
swin_t
,
swin_v2_b
,
swin_v2_s
,
swin_v2_t
from
torchvision.models.vgg
import
vgg11
,
vgg11_bn
,
vgg13
,
vgg13_bn
,
vgg16
,
vgg16_bn
,
vgg19
,
vgg19_bn
from
torchvision.models.vgg
import
vgg11
,
vgg11_bn
,
vgg13
,
vgg13_bn
,
vgg16
,
vgg16_bn
,
vgg19
,
vgg19_bn
from
torchvision.models.vision_transformer
import
vit_b_16
,
vit_b_32
,
vit_h_14
,
vit_l_16
,
vit_l_32
from
torchvision.models.vision_transformer
import
vit_b_16
,
vit_b_32
,
vit_h_14
,
vit_l_16
,
vit_l_32
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment