Function swin_s

Image_Classification/src/models/swin.py:234–236

(num_classes:int, hidden_dim=96, layers=(2, 2, 18, 2), heads=(3, 6, 12, 24), **kwargs)

def swin_s(num_classes:int, hidden_dim=96, layers=(2, 2, 18, 2), heads=(3, 6, 12, 24), **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_b(num_classes:int, hidden_dim=128, layers=(2, 2, 18, 2), heads=(4, 8, 16, 32), **kwargs):