<reponame>sidsrini12/FURL_Sim<filename>LBGM/src/common/config.py
ckpt_dir = '/WorkSpace/FedOptim/ckpts'
data_dir = '/data'
tb_dir = '/WorkSpace/FedOptim/runs'
download = True

num_trains = {
    'celeba': 81385,  # total 162770
    'cifar': 50000,
    'coco': 59144,  # total 118287
    'fmnist': 60000,
    'mnist': 60000,
    'svhn': 73257,
    'voc': 1464,
}

num_tests = {
    'celeba': 9934,  # total 19867
    'cifar': 10000,
    'coco': 5000,  # total 5000
    'fmnist': 10000,
    'mnist': 10000,
    'svhn': 26032,
    'voc': 1449,
}

input_sizes = {
    'celeba': 3*32*32,
    'cifar': 3*32*32,
    'coco': 3*32*32,
    'fmnist': 28*28,
    'mnist': 28*28,
    'svhn': 3*32*32,
    'voc': 3*32*32,
}

output_sizes = {
    'celeba': 10,
    'cifar': 10,
    'coco': 2,
    'fmnist': 10,
    'mnist': 10,
    'svhn': 10,
    'voc': 22,
}

num_channels = {
    'celeba': 3,
    'cifar': 3,
    'coco': 3,
    'fmnist': 1,
    'mnist': 1,
    'svhn': 3,
    'voc': 3,
}

cnn_view = {
    'celeba': 5*5*50,
    'cifar': 5*5*50,
    'coco': 5*5*50,
    'fmnist': 4*4*50,
    'mnist': 4*4*50,
    'svhn': 5*5*50,
    'voc': 5*5*50,
}

im_size = {
    'celeba': 32,
    'coco': 32,
    'voc': 64,
}

model_im_size = {
    'cnn': 0,
    'fcn': 0,
    'resnet18': 0,
    'svm': 0,
    'unet': 0,
    'vgg19': 32,
}
