How to load pretrained weights of swim transformers

1

from torchvision.models import swin_v2_b,swin_t, swin_b
model = swin_b(weights=“Swin_B_Weights”)

It gives error
KeyError: ‘Swin_B_Weights’

2

This should work:

from torchvision.models import swin_v2_b,swin_t, swin_b, Swin_B_Weights
model = swin_b(weights=Swin_B_Weights.IMAGENET1K_V1)