-
Notifications
You must be signed in to change notification settings - Fork 7k
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
Graduate ConvNeXt to main TorchVision area #5330
Conversation
💊 CI failures summary and remediationsAs of commit d40ba15 (more details on the Dr. CI page):
🕵️ 1 new failure recognized by patternsThe following CI failures do not appear to be due to upstream breakages: CodeQL / build (1/1)Step: "Build TorchVision" (full log | diagnosis details | 🔁 rerun)
|
Job | Step | Action |
---|---|---|
cmake_macos_cpu | curl -o conda.sh https://repo.anaconda.com/miniconda/Miniconda3-latest-MacOSX-x86_64.sh | |
sh conda.sh -b | ||
source $HOME/miniconda3/bin/activate | ||
conda install -yq conda-build cmake | ||
packaging/build_cmake.sh | ||
🔁 rerun |
This comment was automatically generated by Dr. CI (expand for details).
Please report bugs/suggestions to the (internal) Dr. CI Users group.
} | ||
|
||
|
||
class LayerNorm2d(nn.LayerNorm): |
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.
Start of copy-pasted code.
return x | ||
|
||
def forward(self, x: Tensor) -> Tensor: | ||
return self._forward_impl(x) |
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.
End of copy-pasted code.
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.
Thanks @datumbox , I only have minor comments, LGTM
`"A ConvNet for the 2020s" <https://arxiv.org/abs/2201.03545>`_ paper. | ||
Args: | ||
pretrained (bool): If True, returns a model pre-trained on ImageNet | ||
progress (bool): If True, displays a progress bar of the download to stderr |
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.
should we document the kwargs here? For the RAFT models I wrote
kwargs (dict): Parameters that will be passed to the :class:`~torchvision.models.optical_flow.RAFT` class
to override any default.
which I would agree is still awkward, considering the RAFT class isn't even exposed in the docs for now
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.
Hmm... Nowhere else on the models
module we use document kwargs
. Perhaps that's something worth considering on the new documentation? I think it might be worth chatting about how we will use it going forwards (previously we pushed parameters that are specific to the model and that kept the model builder signatures uniform, though I'm not sure why this is good).
Summary: * Graduate ConvNeXt to main TorchVision area. * Linter and all var. * Renaming var and making named params mandatory. Reviewed By: kazhang Differential Revision: D33927493 fbshipit-source-id: 12dcbd2463f6dfce7e6b53a347f0c5bfa42b3a48
Moving ConvNeXt's key classes from prototype to main area. The prototype area remains only for the Multi-weights support API.