method: bayes
metric:
  goal: minimize
  name: test_loss
parameters:
  batch_size:
    distribution: int_uniform
    max: 512
    min: 128
  dataset:
    distribution: categorical
    values:
      - igf_1
  epochs:
    distribution: int_uniform
    max: 60
    min: 15
  learning_rate:
    distribution: uniform
    max: 0.02
    min: 0.005
  model_name:
    distribution: categorical
    values:
      - PhaseNet
  pretrained:
    distribution: categorical
    values:
      - diting
      - ethz
      - geofon
      - instance
      - iquique
      - lendb
      - neic
      - original
      - scedc
program: training_wandb_sweep.py