example.json

{
    "dataset": "mnist",
    "tag": "large_test",
    "swarm_config": {
        "number_of_devices": 10,
        "local_set_size": 2,
        "local_data_size": 150,
        "goal_set_size": 5,
        "district_9": [[1,2], [3,4], [5,6],
                       [7,8], [9,0], [1,2],
                       [4,5], [6,7], [8,9]],
        "device_sizes": {
                        "model_sizes": [10, 10, 8, 6, 4],
                        "device_powers": [10, 8, 6, 4, 2]
                        }
    },
    "device_config": {
        "device_strategy": "greedy",
        "id": 0,
        "pretrained_model": "none",
        "model_size": 10,
        "model_size_in_bits": 6224000,
        "device_power": 8,
        "encounter_config": {
            "encounter_data_file": "encounter_data/sigcomm_new_5000.pickle",
            "computation_time": 0.24,
            "communication_rate": 2000000
        },
        "train_config": {
            "__comment__": "formerly called hyperparams",
            "batch-size": 50,
            "orig-lr": 0.1,
            "max_rounds": 3,
            "evaluation-metrics": "loss-and-accuracy",
            "kappa": 7,
            "offset": 0.96,
            "apply-rate": 9,
            "test-data-per-label": 800,
            "similarity-threshold": 0.5,
            "low-similarity-threshold": 0.5,
            "num-rounds": 250,
            "time-per-round": 100,
            "decay": 0.998,
            "optimizer": "adam",
            "optimizer-params": {
                "learning_rate": 0.001
            }
        },
        "__comment__": "This is an examplar json that is sent to simulate_device_services"
    },
    "__comment__": "This is an examplar json that is used by controller"
}