Skip to content

GitLab

Unverified Commit b991bc10 authored by chicm-ms's avatar chicm-ms Committed by GitHub
Browse files

Fix model compression examples (#2557)

parent f7052bda
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
examples/model_compress/model_prune_torch.py
View file @ b991bc10
...@@ -32,7 +32,7 @@ prune_config = { ...@@ -32,7 +32,7 @@ prune_config = {
'start_epoch': 0, 'start_epoch': 0,
'end_epoch': 10, 'end_epoch': 10,
'frequency': 1, 'frequency': 1,
'op_types': ['default'] 'op_types': ['Conv2d']
}] }]
}, },
'slim': { 'slim': {
...@@ -79,7 +79,7 @@ prune_config = { ...@@ -79,7 +79,7 @@ prune_config = {
'pruner_class': ActivationAPoZRankFilterPruner, 'pruner_class': ActivationAPoZRankFilterPruner,
'config_list': [{ 'config_list': [{
'sparsity': 0.5, 'sparsity': 0.5,
'op_types': ['default'], 'op_types': ['Conv2d'],
'op_names': ['feature.0', 'feature.24', 'feature.27', 'feature.30', 'feature.34', 'feature.37'] 'op_names': ['feature.0', 'feature.24', 'feature.27', 'feature.30', 'feature.34', 'feature.37']
}] }]
} }
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment