-
Notifications
You must be signed in to change notification settings - Fork 7.1k
NOMRG Models docs revamp -- v2 #5577
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.
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
Conversation
💊 CI failures summary and remediationsAs of commit 2503263 (more details on the Dr. CI page):
1 failure not recognized by patterns:
This comment was automatically generated by Dr. CI (expand for details).Please report bugs/suggestions to the (internal) Dr. CI Users group. |
@@ -304,36 +305,81 @@ class Wide_ResNet101_2_Weights(WeightsEnum): | |||
DEFAULT = IMAGENET1K_V2 | |||
|
|||
|
|||
# It may or may not be a good idea to somehow unify that with _VGG_DOC . It's | |||
# too early to decide yet. | |||
_RESNET_DOC = """ResNet-{num_layers} from `Deep Residual Learning for Image Recognition <https://arxiv.org/pdf/1512.03385.pdf>`__. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Please ignore _RESNET_DOC
, it's not used anywhere and is replaced by _make_resnet_docstring
. Will remove
allowed_values = ", ".join(f"``{v}``" for v in weights) | ||
|
||
return f"""ResNet-{suffix} from `Deep Residual Learning for Image Recognition <https://arxiv.org/pdf/1512.03385.pdf>`__. | ||
Image Recognition <https://arxiv.org/abs/1409.1556>`__. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
TODO
Image Recognition <https://arxiv.org/abs/1409.1556>`__. |
Follow up to #5575, adding documentation for the Weight enums.
The Weight docs is embedded in the same page as their corresponding model builders: https://1256058-73328905-gh.circle-artifacts.com/0/docs/prototype_models/generated/torchvision.prototype.models.resnet50.html#torchvision.prototype.models.resnet50
