config.py

configs = {
    'cifar10': {
        'num_train_data': 50000,
        'train_bs': 128,
        'test_bs': 512,
        'lr': 0.05,
        'marginal_cost': 1.024e-07,
        'local_steps': 6,
        'random_seed': 1,
        'test_frequency': 500,
        'log_frequency': 60,
        'test_batches': 30,
        'epochs': 100,
        'file_path': 'output',
        'non_iid': False,
        'dirichlet_value': 0.3,
        'uniform_cost': True,
        'name': 'fact-random-sandwich-uniform-cost-iid-run1'
    },

    'mnist': {
            'num_train_data': 60000,
            'train_bs': 128,
            'test_bs': 1024,
            'lr': 5e-4,
            'marginal_cost': 7.11111111111e-08,
            'local_steps': 6,
            'random_seed': 1,
            'log_frequency': 30,
            'test_batches': 30,
            'epochs': 100,
            'file_path': 'output',
            'non_iid': False,
            'dirichlet_value': 0.3,
            'uniform_cost': True,
            'name': 'fact-random-sandwich-uniform-cost-iid-run1'
        },

    'ham10000': {
            'num_train_data': 8012,
            'train_bs': 64,
            'test_bs': 1024,
            'lr': 1e-3,
            'marginal_cost': 7.79e-07,  # gets to 801 data points, 10 workers
            'local_steps': 6,
            'random_seed': 2024,
            'test_frequency': 500,
            'log_frequency': 20,
            'test_batches': 30,
            'epochs': 50,
            'file_path': 'output',
            'non_iid': False,
            'dirichlet_value': 0.3,
            'uniform_cost': False,
            'robustness': {'use': True, 'irrational_agents': 8, 'fr_factor': 0.25},
            'data_path': '../../../../../cmlscratch/marcob/HAM10000',  # 'data/HAM10000',
            'name': 'fact-robustness-8-nonuniform-cost-iid-run1'
        }
}