models.json

{
  "retinanet": {
    "task": "detection",
    "model_space": {
      "accuracy_rating": 8,
      "speed_rating": 2,
      "memory_rating": 4
    },
    "hp_search_space": [
      {
        "name": "learning_rate",
        "values": [
          5e-4,
          1e-5,
          5e-5
        ]
      },
      {
        "name": "anchor_scales",
        "values": [
          [1.0, 1.189207115002721, 1.4142135623730951],
          [1.0, 1.2599210498948732, 1.5874010519681994],
          [1.0, 1.5, 2.0]
        ]
      },
      {
        "name": "anchor_ratios",
        "values": [
          [0.306, 0.638, 1.0, 1.567, 3.266],
          [0.3, 0.5, 1.0, 2.0, 3.0],
          [0.6666666666666666, 0.8, 1.0, 1.25, 1.5]
        ]
      }
    ],
    "training_configs": {
      "hyperparameter_tuner/epochs": 30,
      "depth": 18,
      "input_size": 608,
      "learning_rate": 1e-5,
      "anchor_scales": [1.0, 1.2599210498948732, 1.5874010519681994],
      "anchor_ratios": [0.5, 1.0, 2.0],
      "batch": 1,
      "augment_policy": null
    }
  }
}