maxvit.rst 663 B

1234567891011121314151617181920212223
  1. MaxVit
  2. ===============
  3. .. currentmodule:: torchvision.models
  4. The MaxVit transformer models are based on the `MaxViT: Multi-Axis Vision Transformer <https://arxiv.org/abs/2204.01697>`__
  5. paper.
  6. Model builders
  7. --------------
  8. The following model builders can be used to instantiate an MaxVit model with and without pre-trained weights.
  9. All the model builders internally rely on the ``torchvision.models.maxvit.MaxVit``
  10. base class. Please refer to the `source code
  11. <https://github.com/pytorch/vision/blob/main/torchvision/models/maxvit.py>`_ for
  12. more details about this class.
  13. .. autosummary::
  14. :toctree: generated/
  15. :template: function.rst
  16. maxvit_t