search_space.json

{
    "batch_size": {"_type":"choice", "_value": [16, 32, 64, 128]},
    "lr":{"_type":"choice","_value":[0.0001, 0.001, 0.01, 0.1]},
    "momentum":{"_type":"uniform","_value":[0, 1]}
}