CPU_DATASETS = ['cifar100', 'cifar10', 'waterbirds']
FFCV_DATASETS = ['imagenet', 'dtd', 'flowers102', 'ucf101',
                 'food101', 'sun397', 'oxfordpets', 'stanfordcars']
ALL_DATASETS = CPU_DATASETS + FFCV_DATASETS

CLASS_NUM = {
    'dtd': 47,
    'flowers102': 102,
    'food101': 101,
    'imagenet': 1000,
    'oxfordpets': 37,
    'stanfordcars': 196,
    'sun397': 397,
    'ucf101': 101,
    'cifar10': 10,
    'cifar100': 100,
    'waterbirds': 2
}
