# contains some information on each dataset

datasets_config = {
    "mnist": {
        "real_name": "mnist",
        "nb_classes": 10,
        "image_size": 28,
        "channels": 1,
        "split": 0.2,
    },
    "svhn": {
        "real_name": "svhn_cropped",
        "nb_classes": 10,
        "image_size": 32,
        "channels": 3,
        "split": 0.2,
    },
    "cifar10": {
        "real_name": "cifar10",
        "nb_classes": 10,
        "image_size": 32,
        "channels": 3,
        "split": 0.2,
    },
    "cifar100": {
        "real_name": "cifar100",
        "nb_classes": 100,
        "image_size": 32,
        "channels": 3,
        "split": 0.2,
    },
    "tiny_imagenet": {
        "real_name": "tiny_imagenet",
        "nb_classes": 200,
        "image_size": 64,
        "channels": 3,
        "split": 0.1,
    },
    "fake": {
        "real_name": "fake",
        "nb_classes": 10,
        "image_size": 32,
        "channels": 1,
        "split": 0.2,
    },
    "cifar10_full": {
        "real_name": "cifar10",
        "nb_classes": 10,
        "image_size": 32,
        "channels": 3,
        "split": 0.2,
    },
    "cifar100_full": {
        "real_name": "cifar100",
        "nb_classes": 100,
        "image_size": 32,
        "channels": 3,
        "split": 0.2,
    },
    "tiny_imagenet_full": {
        "real_name": "tiny_imagenet",
        "nb_classes": 200,
        "image_size": 64,
        "channels": 3,
        "split": 0.1,
    }
    
}