Esempio n. 1
0
def get_datasets():
  """Returns all the data sets."""
  return h.sweep(
      "dataset.name",
      h.categorical(["dsprites_full"
          #"dsprites_full", "color_dsprites", "noisy_dsprites",
          #"scream_dsprites", "smallnorb", "cars3d", "shapes3d"
      ]))
Esempio n. 2
0
def get_seeds(num):
    """Returns random seeds."""
    return h.sweep("model.random_seed", h.categorical(list(range(num))))
Esempio n. 3
0
def get_datasets():
    """Returns all the data sets."""
    return h.sweep("dataset.name",
                   h.categorical(["shapes3d", "abstract_dsprites"]))
def get_datasets():
  """Returns all the data sets."""
  return h.sweep(
      "dataset.name",
      h.categorical([
          "mpi3d_toy"]))