Home | 简体中文 | 繁体中文 | 杂文 | Github | 知乎专栏 | Facebook | Linkedin | Youtube | 打赏(Donations) | About
知乎专栏

13.4. Swin Transformer

https://github.com/microsoft/Swin-Transformer https://github.com/ChristophReich1996/Swin-Transformer-V2 https://github.com/SwinTransformer/Swin-Transformer-Object-Detection

13.4.1. 配置权重

https://pytorch.org/vision/master/models/swin_transformer.html

			
from torchvision.models.swin_transformer import swin_b, Swin_B_Weights  
  
model = swin_b(weights=Swin_B_Weights.DEFAULT)