icnet_camvid_magnitude_sparsity_int8.json

{
    "model": "icnet",
    "dataset" : "camvid",

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

    "augmentations": {
        "random_hflip": 0.5
    },


    "input_info": {
      "sample_size": [1, 3, 768, 960]
    },

    "weighing": "mfb",

    "optimizer": {
        "type": "adam",
        "optimizer_params": {
            "lr": 5.0e-5,
            "weight_decay": 2.0e-4
        },
        "schedule_type": "poly",
        "scheduler_params": {
            "power": 0.9
        }
    },

    "model_params": {
         "in_channels": 3,
         "input_size_hw": [768, 960]
    },

    "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",
            "ignored_scopes": ["ICNet/interpolate_0"]
    }
    ]
}