Skip to content

Commit 880fcc3

Browse files
wilxyliukai
authored and
liukai
committed
update configs
1 parent 556c4d2 commit 880fcc3

File tree

2 files changed

+1
-7
lines changed

2 files changed

+1
-7
lines changed

projects/group_fisher/configs/mmdet/group-fisher-finetune_retinanet_resnet50_8xb2_coco.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -15,12 +15,8 @@
1515
mutator=dict(
1616
type='GroupFisherChannelMutator',
1717
parse_cfg=dict(type='ChannelAnalyzer', tracer_type='FxTracer'),
18-
channel_unit_cfg=dict(
19-
type='GroupFisherChannelUnit',
20-
default_args=dict(choice_mode='ratio'))),
18+
channel_unit_cfg=dict(type='GroupFisherChannelUnit')),
2119
init_cfg=dict(type='Pretrained', checkpoint=pruned_path),
2220
)
2321

2422
optim_wrapper = dict(type='OptimWrapper', optimizer=dict(lr=0.01))
25-
26-
find_unused_parameters = True

projects/group_fisher/configs/mmdet/group-fisher-pruning_retinanet_resnet50_8xb2_coco.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -27,5 +27,3 @@
2727
optim_wrapper = dict(
2828
type='OptimWrapper',
2929
optimizer=dict(type='SGD', lr=0.002, momentum=0.9, weight_decay=0.0001))
30-
31-
find_unused_parameters = True

0 commit comments

Comments
 (0)