program: /data/spatop/sweep/run_sweep.py
method: bayes
metric:
  name: validation_accuracy
  goal: maximize
parameters:
  learning_rate:
    distribution: log_uniform_values
    min: 0.0001
    max: 0.001
  hidden_dim:
    values: [64, 128, 256]
  dropout:
    distribution: uniform
    min: 0.1
    max: 0.3
  batch_size:
    values: [512, 1024, 2048]