cfg_cifar10 = {
    "model_flag": 1,
    "delta": 2500,
    "alpha": 1e-1,
    "beta": 0.95,
    "theta": 30,
    "layers_flag": 1,
    "WTA": False,
    "im_scale": [0, 255],
    "scale_flag": True,
    "nd_fixed": False,
    "init_size": 0,
    'vis': False,
    'num_samples': 1,
    "kernel": 0,
    "rho": 0.15
}

cfg_cifar100 = {
    "model_flag": 1,
    "delta": 5000,
    "alpha": 1e-1,
    "beta": 0.95,
    "theta": 30,
    "layers_flag": 1,
    "WTA": False,
    "im_scale": [0, 255],
    "scale_flag": True,
    "nd_fixed": False,
    "init_size": 0,
    'vis': False,
    'num_samples': 1,
    "kernel": 0,
    "rho": 0.15
}

cfg_tiny = {
    "model_flag": 1,
    "delta": 10000,
    "alpha": 1e-1,
    "beta": 0.95,
    "theta": 30,
    "layers_flag": 1,
    "WTA": False,
    "im_scale": [0, 255],
    "scale_flag": True,
    "nd_fixed": False,
    "init_size": 0,
    'vis': False,
    'num_samples': 1,
    "kernel": 0,
    "rho": 0.15
}