video_swin_transformer.rst 778 B

123456789101112131415161718192021222324252627
  1. Video SwinTransformer
  2. =====================
  3. .. currentmodule:: torchvision.models.video
  4. The Video SwinTransformer model is based on the `Video Swin Transformer <https://arxiv.org/abs/2106.13230>`__ paper.
  5. .. betastatus:: video module
  6. Model builders
  7. --------------
  8. The following model builders can be used to instantiate a VideoResNet model, with or
  9. without pre-trained weights. All the model builders internally rely on the
  10. ``torchvision.models.video.swin_transformer.SwinTransformer3d`` base class. Please refer to the `source
  11. code
  12. <https://github.com/pytorch/vision/blob/main/torchvision/models/video/swin_transformer.py>`_ for
  13. more details about this class.
  14. .. autosummary::
  15. :toctree: generated/
  16. :template: function.rst
  17. swin3d_t
  18. swin3d_s
  19. swin3d_b