update configs

open-mmlab · Jan 16, 2023 · 880fcc3 · 880fcc3
1 parent 556c4d2
commit 880fcc3
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 7 deletions.
diff --git a/projects/group_fisher/configs/mmdet/group-fisher-finetune_retinanet_resnet50_8xb2_coco.py b/projects/group_fisher/configs/mmdet/group-fisher-finetune_retinanet_resnet50_8xb2_coco.py
@@ -15,12 +15,8 @@
     mutator=dict(
         type='GroupFisherChannelMutator',
         parse_cfg=dict(type='ChannelAnalyzer', tracer_type='FxTracer'),
-        channel_unit_cfg=dict(
-            type='GroupFisherChannelUnit',
-            default_args=dict(choice_mode='ratio'))),
+        channel_unit_cfg=dict(type='GroupFisherChannelUnit')),
     init_cfg=dict(type='Pretrained', checkpoint=pruned_path),
 )
 
 optim_wrapper = dict(type='OptimWrapper', optimizer=dict(lr=0.01))
-
-find_unused_parameters = True
diff --git a/projects/group_fisher/configs/mmdet/group-fisher-pruning_retinanet_resnet50_8xb2_coco.py b/projects/group_fisher/configs/mmdet/group-fisher-pruning_retinanet_resnet50_8xb2_coco.py
@@ -27,5 +27,3 @@
 optim_wrapper = dict(
     type='OptimWrapper',
     optimizer=dict(type='SGD', lr=0.002, momentum=0.9, weight_decay=0.0001))
-
-find_unused_parameters = True