Skip to content

Commit

Permalink
Expose raft_large and raft_small to torchhub (pytorch#5174)
Browse files Browse the repository at this point in the history
NicolasHug authored Jan 7, 2022

Verified

This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
1 parent 5e26458 commit a47c46c
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions hubconf.py
Original file line number Diff line number Diff line change
@@ -19,6 +19,7 @@
from torchvision.models.mnasnet import mnasnet0_5, mnasnet0_75, mnasnet1_0, mnasnet1_3
from torchvision.models.mobilenetv2 import mobilenet_v2
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 (
regnet_y_400mf,
regnet_y_800mf,

0 comments on commit a47c46c

Please sign in to comment.