| { | |
| "compression": [ | |
| { | |
| "algorithm": "movement_sparsity", | |
| "ignored_scopes": [ | |
| "{re}.*PatchEmbed.*", | |
| "{re}.*PatchMerging.*", | |
| "{re}.*classifier.*", | |
| "{re}.*LayerNorm.*" | |
| ], | |
| "params": { | |
| "enable_structured_masking": true, | |
| "importance_regularization_factor": 1.0, | |
| "warmup_end_epoch": 5, | |
| "warmup_start_epoch": 2 | |
| }, | |
| "sparse_structure_by_scopes": [ | |
| { | |
| "mode": "block", | |
| "sparse_factors": [ | |
| 16, | |
| 16 | |
| ], | |
| "target_scopes": "{re}.*SwinAttention.*" | |
| }, | |
| { | |
| "axis": 0, | |
| "mode": "per_dim", | |
| "target_scopes": "{re}.*SwinIntermediate.*" | |
| }, | |
| { | |
| "axis": 1, | |
| "mode": "per_dim", | |
| "target_scopes": "{re}.*SwinOutput.*" | |
| } | |
| ] | |
| }, | |
| { | |
| "algorithm": "quantization", | |
| "export_to_onnx_standard_ops": false, | |
| "ignored_scopes": [ | |
| "{re}.*__add___[0-1]", | |
| "{re}.*layer_norm_0", | |
| "{re}.*matmul_1", | |
| "{re}.*__truediv__*" | |
| ], | |
| "initializer": { | |
| "batchnorm_adaptation": { | |
| "num_bn_adaptation_samples": 200 | |
| }, | |
| "range": { | |
| "num_init_samples": 32, | |
| "params": { | |
| "max_percentile": 99.99, | |
| "min_percentile": 0.01 | |
| }, | |
| "type": "percentile" | |
| } | |
| }, | |
| "overflow_fix": "enable", | |
| "preset": "mixed", | |
| "scope_overrides": { | |
| "activations": { | |
| "{re}.*matmul_0": { | |
| "mode": "symmetric" | |
| } | |
| } | |
| } | |
| } | |
| ], | |
| "input_info": [ | |
| { | |
| "keyword": "pixel_values", | |
| "sample_size": [ | |
| 16, | |
| 3, | |
| 224, | |
| 224 | |
| ], | |
| "type": "float" | |
| } | |
| ], | |
| "optimum_version": "1.7.1", | |
| "save_onnx_model": false, | |
| "transformers_version": "4.26.1" | |
| } | |