hyperparameter_search_config.json

{
  "epochs": {
    "min": 1,
    "max": 50
  },
  "batch_size": {
    "values": [
      8,
      16,
      32
    ]
  },
  "learning_rate": {
    "distribution": "log_uniform_values",
    "min": 1e-5,
    "max": 1e-3
  },
  "seed": {
    "min": 1,
    "max": 100
  },
  "weight_decay": {
    "values": [
      0.0,
      0.1,
      0.2,
      0.3,
      0.4,
      0.5
    ]
  }
}