File size: 812 Bytes
663f02a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 |
{
"input_info": [
{
"sample_size": [1, 384],
"type": "long"
},
{
"sample_size": [1, 384],
"type": "long"
},
{
"sample_size": [1, 384],
"type": "long"
}
],
"compression": {
"algorithm": "movement_sparsity",
"params": {
"schedule": "poly_threshold",
"power": 3,
"init_importance_threshold": 0.0,
"final_importance_threshold": 0.1,
"warmup_start_epoch": 1,
"warmup_end_epoch": 2.5,
"steps_per_epoch": 5533,
"regu_final_lambda": 400,
"update_per_optimizer_step": true,
},
"ignored_scopes": ["{re}.*NNCFEmbedding", "{re}.*qa_outputs*"]
},
}
|