File size: 359 Bytes
68b32f4 |
1 2 3 4 5 6 7 8 9 10 11 |
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'
]
|