Skip to content

GitLab

Unverified Commit a47c46cb authored by Nicolas Hug's avatar Nicolas Hug Committed by GitHub
Browse files

Expose raft_large and raft_small to torchhub (#5174)

parent 5e264580
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
hubconf.py
View file @ a47c46cb
...@@ -19,6 +19,7 @@ from torchvision.models.inception import inception_v3 ...@@ -19,6 +19,7 @@ from torchvision.models.inception import inception_v3
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
from torchvision.models.optical_flow import raft_large, raft_small
from torchvision.models.regnet import ( from torchvision.models.regnet import (
regnet_y_400mf, regnet_y_400mf,
regnet_y_800mf, regnet_y_800mf,
......
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