{ "type" : ["str", "multistep"], "batch_size": ["int", 128], "epochs" : ["int", 300], "momentum" : ["float", 0.9], "decay" : ["float", 0.0001], "LR" : ["float", 0.5], "milestones": ["int", [150, 225]], "gammas" : ["float", [0.1, 0.1]] }