resnet50_imagenet_rb_sparsity.json

{
    "model": "ResNet50",
    "pretrained": true,

    "input_info": {
      "sample_size": [1, 224, 224, 3]
    },
    "batch_size" : 256,
    "epochs": 55,
    "optimizer": {
        "type": "Adam",
        "schedule_type": "piecewise_constant",
        "schedule_params": {
            "boundaries": [20, 30, 40, 45],
            "values": [1e-3, 5e-4, 1e-4, 1e-5, 1e-6]
        }
    },

    "dataset": "imagenet2012",
    "dataset_type": "tfds",

    "compression": {
        "algorithm": "rb_sparsity",
        "sparsity_init": 0.02,
        "params": {
            "sparsity_target": 0.8,
            "sparsity_target_epoch": 18,
            "sparsity_freeze_epoch": 35
        }
    }
}