Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Expose on Hub the public methods of the registration API #6364

Merged
merged 4 commits into from
Aug 8, 2022
Merged

Expose on Hub the public methods of the registration API #6364

Changes from 1 commit
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
cdf8b83
Expose on Hub the public methods of the registration API
datumbox Aug 3, 2022
484dc7b
Limit methods and update docs.
datumbox Aug 3, 2022
0530f22
Merge branch 'main' into hub/expose_functions
datumbox Aug 4, 2022
daccbf1
Merge branch 'main' into hub/expose_functions
datumbox Aug 8, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion hubconf.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
# Optional list of dependencies required by the package
dependencies = ["torch"]

from torchvision.models import get_weight
from torchvision.models import get_model, get_model_weights, get_weight, list_models
from torchvision.models.alexnet import alexnet
from torchvision.models.convnext import convnext_base, convnext_large, convnext_small, convnext_tiny
from torchvision.models.densenet import densenet121, densenet161, densenet169, densenet201
Expand Down