config.json

{
  "activation": "relu",
  "base_features": 64,
  "batch_norm": true,
  "batch_size": 16,
  "criterion": "bce",
  "img_size": 224,
  "lr": 0.004,
  "max_epochs": 50,
  "num_predictions": 68,
  "optimizer": "adam",
  "output_size": 128,
  "patience": 10,
  "random_seed": 42,
  "use_scheduler": true
}