Function swin_b

Image_Classification/src/models/swin.py:239–241

(num_classes:int, hidden_dim=128, layers=(2, 2, 18, 2), heads=(4, 8, 16, 32), **kwargs)

def swin_b(num_classes:int, hidden_dim=128, layers=(2, 2, 18, 2), heads=(4, 8, 16, 32), **kwargs):
    return SwinTransformer(num_classes=num_classes, hidden_dim=hidden_dim, layers=layers, heads=heads,
                           downscaling_factors=(2, 2, 2, 1), window_size=4, **kwargs)


def swin_l(num_classes:int, hidden_dim=192, layers=(2, 2, 18, 2), heads=(6, 12, 24, 48), **kwargs):