sweep.yaml

program: train_rave.py
method: random
metric:
  goal: minimize
  name:
command:
  - python3
  - ${program}
  - ${args}
parameters:
  batch_size:
    values:
      - 8
      - 16
      - 32
      - 64
    distribution: categorical
  accum_batches:
    values:
      - 4
      - 8
      - 16
    distribution: categorical
  latent_size:
    values:
      - 128
      - 256
      - 512
      - 1024
      - 2048
    distribution: categorical
  data_size:
    values:
      - 8
      - 16
      - 32
      - 64
    distribution: categorical
  capacity:
    values:
      - 32
      - 64
      - 96
      - 128
    distribution: categorical
  warmup:
    max: 2000000
    min: 100000
    distribution: int_uniform