SwinTransformer =============== .. currentmodule:: torchvision.models The SwinTransformer models are based on the `Swin Transformer: Hierarchical Vision Transformer using Shifted Windows `__ paper. SwinTransformer V2 models are based on the `Swin Transformer V2: Scaling Up Capacity and Resolution `__ paper. Model builders -------------- The following model builders can be used to instantiate an SwinTransformer model (original and V2) with and without pre-trained weights. All the model builders internally rely on the ``torchvision.models.swin_transformer.SwinTransformer`` base class. Please refer to the `source code `_ for more details about this class. .. autosummary:: :toctree: generated/ :template: function.rst swin_t swin_s swin_b swin_v2_t swin_v2_s swin_v2_b