pruning_config.json

{
  "explicit_prune": {},
  "init_strategy": "uniform",
  "initial_step": 0,
  "not_to_prune": [
    "qa_outputs",
    "pooler",
    "teacher"
  ],
  "prune_layer_types": {
    "Linear": {
      "name": "weight"
    },
    "QuantizedLinear": {
      "name": "weight"
    }
  },
  "pruning_fn": "pattern_lock",
  "pruning_fn_default_kwargs": {},
  "scheduler": "one_shot",
  "weight_sparsity_map": {}
}