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

Jump to bottom

Code block in Using models from Hub failed to render correctly. #8837

Open
GdoongMathew opened this issue Jan 5, 2025 · 1 comment · May be fixed by #8846
Open

Code block in Using models from Hub failed to render correctly. #8837

GdoongMathew opened this issue Jan 5, 2025 · 1 comment · May be fixed by #8846

Comments

@GdoongMathew
Copy link

GdoongMathew commented Jan 5, 2025 •
edited
Loading

📚 The doc issue

In https://pytorch.org/vision/main/models.html#using-models-from-hub, the resulting code block on my browser looks like
image
, which is different from the original documents.

import torch

# Option 1: passing weights param as string
model = torch.hub.load("pytorch/vision", "resnet50", weights="IMAGENET1K_V2")

# Option 2: passing weights param as enum
weights = torch.hub.load("pytorch/vision", "get_weight", weights="ResNet50_Weights.IMAGENET1K_V2")  # --> this is where it looks weird on Chrome.
model = torch.hub.load("pytorch/vision", "resnet50", weights=weights)

Suggest a potential alternative/fix

Not sure if the following changes could solve the rendering problem.

# Option 2: passing weights param as enum
weights = torch.hub.load(
    "pytorch/vision", 
    "get_weight", 
    weights="ResNet50_Weights.IMAGENET1K_V2",
)
@NicolasHug
Copy link
Member

Thanks for the report @GdoongMathew , your suggestion sounds good, would you mind submitting a PR for this?

@GdoongMathew GdoongMathew linked a pull request Jan 10, 2025 that will close this issue
Open
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging a pull request may close this issue.

fix: fix code block rendering problem in Using models from Hub GdoongMathew/vision
2 participants
@NicolasHug @GdoongMathew