models/constants.py

VALID_NEURON_SELECT_TYPES = ['first-last', 'random', 'random-pairing']

VALID_BACKBONE_TYPES = [
    f'resnet{depth}-{i}' for depth in [18, 34, 50, 101, 152] for i in range(1, 5)
] + ['shallow-wide', 'parity_backbone']

VALID_POSITIONAL_EMBEDDING_TYPES = [
    'learnable-fourier', 'multi-learnable-fourier',
    'custom-rotational', 'custom-rotational-1d'
]