{
    "model": "RetinaNet",
    "input_info": {
        "sample_size": [1, 640, 640, 3]
    },

    "epochs": 20,
    "batch_size": 64,
    "weight_decay": 0.0001,

    "dataset": "coco/2017",
    "dataset_type": "tfds",

    "optimizer": {
        "type": "momentum",
        "optimizer_params": {
            "momentum": 0.9,
            "nesterov": true
        },
        "schedule_type": "step_warmup",
        "schedule_params": {
            "warmup_learning_rate": 0.0067,
            "warmup_steps": 500,
            "init_learning_rate": 0.08,
            "learning_rate_levels": [
                0.008,
                0.0008,
                0.00008
            ],
            "learning_rate_steps": [
                23000,
                31000,
                37000
            ]
        }
    },

    "compression": [
        {
            "algorithm": "filter_pruning",
            "pruning_init": 0.1,
            "params": {
                "schedule": "exponential",
                "pruning_target": 0.4,
                "pruning_steps": 6,
                "filter_importance": "geometric_median"
            }
        },
        {
            "algorithm": "quantization"
        }
    ]
}