Revamping our classification models docs #5833
Description
We're re-writing our models docs to make them clearer, simpler, and to properly document the upcoming multi-weight API. This issue is about adding docs for the classification models.
Perhaps this is something @oke-aditya @YosuaMichael @lezwon @zhiqwang would be interested in :) ?
Our latest new docs are currently here (this link is likely outdated by the time you look at it, but it doesn't matter; the skeleton is there). We created a separate section that will eventually be merged into the main one. We have documented a few models, but most of them are still missing. The list of models that still need docs is listed below. If you'd like to participate, please comment below with a message saying "I'm working on XYZ" where XYZ is a model, so that others don't pick the same as you do. To keep things simple, please submit one PR per model, but feel free to contribute more than one model.
How to write docs for a model
Note: below are detailed instructions. This makes it look more complicated than it actually is. Don't be scared!
A great place to start is to look a the changes in this PR that documents SqueezeNet. You'll need to do exactly the same for your model:
- Create a new
.rstfile in https://github.com/pytorch/vision/tree/main/docs/source/models. The file should look like this, with a link to the original paper, and a list of the corresponding model builders. It should also mention the base model class and link to the.pyfile where it is defined. - Update the list in https://github.com/pytorch/vision/blame/main/docs/source/models_new.rst to link to this new file (without the
.rstsuffix). Please keep the list alphabetically sorted - Update the docstring of each new model builder, similarly to this one.
- there is a 1:1 mapping between a model builder and a Weight enum. For example, the docstring of squeezenet1_0 makes direct references to SqueezeNet1_0_Weights.
- don't forget the autoclass directive in the docstring. This will auto-generate documentation for the Weight enums. You don't need to understand how this is done, but if you're curious, it's done here.
To build the docs locally, please look at our contributing guide. You won't need to worry about the gallery example, so always use make html-noplot instead of make html to save time.
Please don't hesitate to ping us if you need any help / guidance or if you have any question!
Classification models that need docs are:
- Alexnet - AlexNet New Documentation. #5886
- ConvNeXt - Add docs for ConvNext #5869
- Inception v3 - Add inception docs #5924
- DenseNet - Added docs for DenseNet #5878
- EfficientNet - EfficientNet New Documentation #5854
- EfficientNet v2 - Efficientnetv2 #5864
- GoogLeNet - GoogLeNet New Documentation #5885
- MNASNet - updating mnasnet docs #5920
- MobileNet v2 - docs: Added MobileNetV2 to the new doc #5899
- MobileNet v3 - Add docs for Mobilenetv3 #5908
- RegNet - Adding revamp docs for vision_transformers and regnet #5856
- ResNext - Add docs for ResNeXt #5871
- ShuffleNet v2 - Add revamped docs for
ShuffleNet V2#5921 - VisionTransformer - Adding revamp docs for vision_transformers and regnet #5856
- Wide ResNet - revamp docs for Wide ResNet #5907