[Fix]Dcff Deploy Revision (#383)

* dcff deploy revision * tempsave * update fix_subnet * update mutator load * export/load_fix_subnet revision for mutator * update fix_subnet with dev-1.x * update comments * update docs * update registry
open-mmlab · Dec 16, 2022 · 82e9549 · 82e9549
1 parent 42e8de7
commit 82e9549
Show file tree

Hide file tree

Showing 27 changed files with 1,057 additions and 2,118 deletions.
diff --git a/configs/pruning/mmcls/dcff/dcff_compact_resnet_8xb32_in1k.py b/configs/pruning/mmcls/dcff/dcff_compact_resnet_8xb32_in1k.py
@@ -1,5 +1,13 @@
 _base_ = ['dcff_resnet_8xb32_in1k.py']
 
 # model settings
-model = _base_.model
-model['is_deployed'] = True
+model_cfg = dict(
+    _scope_='mmrazor',
+    type='sub_model',
+    cfg=dict(
+        cfg_path='mmcls::resnet/resnet50_8xb32_in1k.py', pretrained=False),
+    fix_subnet='configs/pruning/mmcls/dcff/fix_subnet.json',
+    mode='mutator',
+    init_cfg=dict(
+        type='Pretrained',
+        checkpoint='configs/pruning/mmcls/dcff/fix_subnet_weight.pth'))
diff --git a/configs/pruning/mmcls/dcff/dcff_resnet_8xb32_in1k.py b/configs/pruning/mmcls/dcff/dcff_resnet_8xb32_in1k.py
@@ -76,7 +76,10 @@
             type='ChannelAnalyzer',
             demo_input=(1, 3, 224, 224),
             tracer_type='BackwardTracer')),
+    fix_subnet=None,
+    data_preprocessor=None,
     target_pruning_ratio=target_pruning_ratio,
     step_freq=1,
-    linear_schedule=False,
-    is_deployed=False)
+    linear_schedule=False)
+
+val_cfg = dict(_delete_=True, type='mmrazor.ItePruneValLoop')
diff --git a/configs/pruning/mmcls/dcff/fix_subnet.json b/configs/pruning/mmcls/dcff/fix_subnet.json
@@ -0,0 +1,141 @@
+{
+    "type":"DCFFChannelMutator",
+    "channel_unit_cfg":{
+        "type":"DCFFChannelUnit",
+        "default_args":{
+            "choice_mode":"ratio"
+        },
+        "units":{
+            "backbone.conv1_(0, 64)_64":{
+                "init_args":{
+                    "num_channels":64,
+                    "choice_mode":"ratio",
+                    "divisor":1,
+                    "min_value":1,
+                    "min_ratio":0.9
+                },
+                "choice":1.0
+            },
+            "backbone.layer1.0.conv1_(0, 64)_64":{
+                "init_args":{
+                    "num_channels":64,
+                    "choice_mode":"ratio",
+                    "divisor":1,
+                    "min_value":1,
+                    "min_ratio":0.9
+                },
+                "choice":0.640625
+            },
+            "backbone.layer1.1.conv1_(0, 64)_64":{
+                "init_args":{
+                    "num_channels":64,
+                    "choice_mode":"ratio",
+                    "divisor":1,
+                    "min_value":1,
+                    "min_ratio":0.9
+                },
+                "choice":0.640625
+            },
+            "backbone.layer2.0.conv1_(0, 128)_128":{
+                "init_args":{
+                    "num_channels":128,
+                    "choice_mode":"ratio",
+                    "divisor":1,
+                    "min_value":1,
+                    "min_ratio":0.9
+                },
+                "choice":0.6484375
+            },
+            "backbone.layer2.0.conv2_(0, 128)_128":{
+                "init_args":{
+                    "num_channels":128,
+                    "choice_mode":"ratio",
+                    "divisor":1,
+                    "min_value":1,
+                    "min_ratio":0.9
+                },
+                "choice":0.59375
+            },
+            "backbone.layer2.1.conv1_(0, 128)_128":{
+                "init_args":{
+                    "num_channels":128,
+                    "choice_mode":"ratio",
+                    "divisor":1,
+                    "min_value":1,
+                    "min_ratio":0.9
+                },
+                "choice":0.6484375
+            },
+            "backbone.layer3.0.conv1_(0, 256)_256":{
+                "init_args":{
+                    "num_channels":256,
+                    "choice_mode":"ratio",
+                    "divisor":1,
+                    "min_value":1,
+                    "min_ratio":0.9
+                },
+                "choice":0.6484375
+            },
+            "backbone.layer3.0.conv2_(0, 256)_256":{
+                "init_args":{
+                    "num_channels":256,
+                    "choice_mode":"ratio",
+                    "divisor":1,
+                    "min_value":1,
+                    "min_ratio":0.9
+                },
+                "choice":0.59765625
+            },
+            "backbone.layer3.1.conv1_(0, 256)_256":{
+                "init_args":{
+                    "num_channels":256,
+                    "choice_mode":"ratio",
+                    "divisor":1,
+                    "min_value":1,
+                    "min_ratio":0.9
+                },
+                "choice":0.6484375
+            },
+            "backbone.layer4.0.conv1_(0, 512)_512":{
+                "init_args":{
+                    "num_channels":512,
+                    "choice_mode":"ratio",
+                    "divisor":1,
+                    "min_value":1,
+                    "min_ratio":0.9
+                },
+                "choice":0.69921875
+            },
+            "backbone.layer4.0.conv2_(0, 512)_512":{
+                "init_args":{
+                    "num_channels":512,
+                    "choice_mode":"ratio",
+                    "divisor":1,
+                    "min_value":1,
+                    "min_ratio":0.9
+                },
+                "choice":0.69921875
+            },
+            "backbone.layer4.1.conv1_(0, 512)_512":{
+                "init_args":{
+                    "num_channels":512,
+                    "choice_mode":"ratio",
+                    "divisor":1,
+                    "min_value":1,
+                    "min_ratio":0.9
+                },
+                "choice":0.69921875
+            }
+        }
+    },
+    "parse_cfg":{
+        "type":"ChannelAnalyzer",
+        "demo_input":[
+            1,
+            3,
+            224,
+            224
+        ],
+        "tracer_type":"BackwardTracer"
+    }
+}