Fix model compression examples (#2557)

b991bc10 · chicm-ms · GitHub · f7052bda · b991bc10
Unverified Commit b991bc10 authored Jun 16, 2020 by chicm-ms Committed by GitHub Jun 16, 2020
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

examples/model_compress/model_prune_torch.py examples/model_compress/model_prune_torch.py +2 -2

No files found.
--- a/examples/model_compress/model_prune_torch.py
+++ b/examples/model_compress/model_prune_torch.py
@@ -32,7 +32,7 @@ prune_config = {
            'start_epoch': 0,
            'end_epoch': 10,
            'frequency': 1,
-            'op_types': ['default']
+            'op_types': ['Conv2d']
        }]
    },
    'slim': {
@@ -79,7 +79,7 @@ prune_config = {
        'pruner_class': ActivationAPoZRankFilterPruner,
        'config_list': [{
            'sparsity': 0.5,
-            'op_types': ['default'],
+            'op_types': ['Conv2d'],
            'op_names': ['feature.0', 'feature.24', 'feature.27', 'feature.30', 'feature.34', 'feature.37']
        }]
    }