unet_camvid_magnitude_sparsity_int8.json

{
    "model": "unet",
    "dataset" : "camvid",

    "preprocessing": {
        "resize": {
            "height": 368,
            "width": 480
        },
        "normalize":
        {
            "mean": [0.39068785, 0.40521392, 0.41434407],
            "std" : [0.29652068, 0.30514979, 0.30080369]
        }
    },


    "input_info": {
      "sample_size": [1, 3, 368, 480]
    },

    "weighing": "mfb",

    "optimizer": {
        "type": "Adam",
        "optimizer_params": {
            "lr": 5.0e-5,
            "weight_decay": 2.0e-4
        },
        "schedule_type": "plateau",
        "scheduler_params": {
            "threshold": 0.002,
            "patience": 3
        },
        "gamma": 0.1
    },

    "model_params": {
         "input_size_hw": [368, 480]
    },
    "compression": [
        {
            "algorithm": "magnitude_sparsity",
            "sparsity_init": 0.1,
            "params": {
                "schedule": "multistep",
                "multistep_steps": [
                    5,
                    10,
                    20,
                    30,
                    40
                ],
                "multistep_sparsity_levels": [
                    0.1,
                    0.2,
                    0.3,
                    0.4,
                    0.5,
                    0.6
                ]
            }
        },
        {
            "algorithm": "quantization"
    }
    ]
}