xtrack2_spearmint.json

{
    "language"        : "PYTHON",
    "experiment-name" : "xtrack2spear",
    "variables" : {
        "n_cells" : {
            "type" : "int",
            "size" : 1,
            "min": 5,
            "max": 30
        },
        "n_epochs" : {
            "type": "int",
            "size": 1,
            "min": 200,
            "max": 200
        },
        "emb_size" : {
            "type" : "int",
            "size" : 1,
            "min": 5,
            "max": 20
        },
        "opt_type" : {
            "type" : "ENUM",
            "size" : 1,
            "options": ["sgd"]
        },
        "lr" : {
            "type" : "float",
            "size" : 1,
            "min": 0.01,
            "max": 0.3
        },
        "p_drop" : {
            "type" : "float",
            "size" : 1,
            "min": 0.0,
            "max": 0.5
        },
        "mb_size" : {
            "type" : "int",
            "size" : 1,
            "min": 1,
            "max": 20
        }
    },
    "database": {
        "address": "10.10.24.131"
    },
    "resources" : {
        "cluster" : {
            "scheduler"         : "SGE",
            "max-concurrent"    : 500,
            "max-finished-jobs" : 10000
        }
    },
    "tasks": {
        "xtrack2" : {
            "type": "OBJECTIVE",
            "likelihood" : "GAUSSIAN",
            "main-file"       : "xtrack2_spearmint.py",
            "resources"  : ["cluster"]
        }
    }
}