"git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "f8d4a1e2833191328a6572890ae15bca793eef7c"
Unverified Commit 703f2b90 authored by Abhijit Deo's avatar Abhijit Deo Committed by GitHub
Browse files

Add revamped docs for ResNeXt (#5871)



* initial commit

* formatting

* minor change

* replaced pdf link with webpage link

* replaced pdf link with webpage link

* minor

* minor
Co-authored-by: default avatarNicolas Hug <contact@nicolas-hug.com>
parent f1967823
ResNeXt
=======
.. currentmodule:: torchvision.models
The ResNext model is based on the `Aggregated Residual Transformations for Deep Neural Networks <https://arxiv.org/abs/1611.05431v2>`__
paper.
Model builders
--------------
The following model builders can be used to instantiate an ResNext model, with or
without pre-trained weights. All the model builders internally rely on the
``torchvision.models.resnet.ResNet`` base class. Please refer to the `source
code
<https://github.com/pytorch/vision/blob/main/torchvision/models/resnet.py>`_ for
more details about this class.
.. autosummary::
:toctree: generated/
:template: function.rst
resnext50_32x4d
resnext101_32x8d
...@@ -41,6 +41,7 @@ weights: ...@@ -41,6 +41,7 @@ weights:
models/efficientnetv2 models/efficientnetv2
models/regnet models/regnet
models/resnet models/resnet
models/resnext
models/squeezenet models/squeezenet
models/vgg models/vgg
models/vision_transformer models/vision_transformer
......
...@@ -680,12 +680,23 @@ def resnet152(*, weights: Optional[ResNet152_Weights] = None, progress: bool = T ...@@ -680,12 +680,23 @@ def resnet152(*, weights: Optional[ResNet152_Weights] = None, progress: bool = T
def resnext50_32x4d( def resnext50_32x4d(
*, weights: Optional[ResNeXt50_32X4D_Weights] = None, progress: bool = True, **kwargs: Any *, weights: Optional[ResNeXt50_32X4D_Weights] = None, progress: bool = True, **kwargs: Any
) -> ResNet: ) -> ResNet:
r"""ResNeXt-50 32x4d model from """ResNeXt-50 32x4d model from
`"Aggregated Residual Transformation for Deep Neural Networks" <https://arxiv.org/pdf/1611.05431.pdf>`_. `Aggregated Residual Transformation for Deep Neural Networks <https://arxiv.org/abs/1611.05431>`_.
Args: Args:
weights (ResNeXt50_32X4D_Weights, optional): The pretrained weights for the model weights (:class:`~torchvision.models.ResNeXt50_32X4D_Weights`, optional): The
progress (bool): If True, displays a progress bar of the download to stderr pretrained weights to use. See
:class:`~torchvision.models.ResNext50_32X4D_Weights` below for
more details, and possible values. By default, no pre-trained
weights are used.
progress (bool, optional): If True, displays a progress bar of the
download to stderr. Default is True.
**kwargs: parameters passed to the ``torchvision.models.resnet.ResNet``
base class. Please refer to the `source code
<https://github.com/pytorch/vision/blob/main/torchvision/models/resnet.py>`_
for more details about this class.
.. autoclass:: torchvision.models.ResNeXt50_32X4D_Weights
:members:
""" """
weights = ResNeXt50_32X4D_Weights.verify(weights) weights = ResNeXt50_32X4D_Weights.verify(weights)
...@@ -698,12 +709,23 @@ def resnext50_32x4d( ...@@ -698,12 +709,23 @@ def resnext50_32x4d(
def resnext101_32x8d( def resnext101_32x8d(
*, weights: Optional[ResNeXt101_32X8D_Weights] = None, progress: bool = True, **kwargs: Any *, weights: Optional[ResNeXt101_32X8D_Weights] = None, progress: bool = True, **kwargs: Any
) -> ResNet: ) -> ResNet:
r"""ResNeXt-101 32x8d model from """ResNeXt-101 32x8d model from
`"Aggregated Residual Transformation for Deep Neural Networks" <https://arxiv.org/pdf/1611.05431.pdf>`_. `Aggregated Residual Transformation for Deep Neural Networks <https://arxiv.org/abs/1611.05431>`_.
Args: Args:
weights (ResNeXt101_32X8D_Weights, optional): The pretrained weights for the model weights (:class:`~torchvision.models.ResNeXt101_32X8D_Weights`, optional): The
progress (bool): If True, displays a progress bar of the download to stderr pretrained weights to use. See
:class:`~torchvision.models.ResNeXt101_32X8D_Weights` below for
more details, and possible values. By default, no pre-trained
weights are used.
progress (bool, optional): If True, displays a progress bar of the
download to stderr. Default is True.
**kwargs: parameters passed to the ``torchvision.models.resnet.ResNet``
base class. Please refer to the `source code
<https://github.com/pytorch/vision/blob/main/torchvision/models/resnet.py>`_
for more details about this class.
.. autoclass:: torchvision.models.ResNeXt101_32X8D_Weights
:members:
""" """
weights = ResNeXt101_32X8D_Weights.verify(weights) weights = ResNeXt101_32X8D_Weights.verify(weights)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment