convnext.rst 692 B

1234567891011121314151617181920212223242526
  1. ConvNeXt
  2. ========
  3. .. currentmodule:: torchvision.models
  4. The ConvNeXt model is based on the `A ConvNet for the 2020s
  5. <https://arxiv.org/abs/2201.03545>`_ paper.
  6. Model builders
  7. --------------
  8. The following model builders can be used to instantiate a ConvNeXt model, with or
  9. without pre-trained weights. All the model builders internally rely on the
  10. ``torchvision.models.convnext.ConvNeXt`` base class. Please refer to the `source code
  11. <https://github.com/pytorch/vision/blob/main/torchvision/models/convnext.py>`_ for
  12. more details about this class.
  13. .. autosummary::
  14. :toctree: generated/
  15. :template: function.rst
  16. convnext_tiny
  17. convnext_small
  18. convnext_base
  19. convnext_large