I have a ray tune configuration which is specified as follows:
config={
"params.central_layer_sizes:": tune.grid_search([[1600,800], [403, 200], [400, 400]]),
"params.right_layer_sizes": tune.grid_search([[48,24], [12, 6], [12, 12]]),
"params.joint_layer_sizes": tune.grid_search([[800,400], [400, 200], [400, 400]])
}
I would like to sample in a concerted way i.e. the index 0 of the three variables go together, index 1 and index 2. Is there a way to specify such grouping out of the box in ray? At the moment, everything is sampled independent of the other and generating configurations that I do not want to test.