Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Fix]Dcff Deploy Revision #383

Merged
merged 12 commits into from
Dec 16, 2022
Merged

[Fix]Dcff Deploy Revision #383

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
12 commits
Select commit Hold shift + click to select a range
3fd9c9b
dcff deploy revision
spynccat Dec 6, 2022
a7e5336
Merge branch 'dev-1.x' into dcff_deploy_revision
spynccat Dec 6, 2022
fb70f78
tempsave
spynccat Dec 9, 2022
69c7d98
Merge branch 'dev-1.x' into dcff_deploy_revision
spynccat Dec 9, 2022
f99de08
update fix_subnet
spynccat Dec 12, 2022
5206866
update mutator load
spynccat Dec 14, 2022
8213b33
export/load_fix_subnet revision for mutator
spynccat Dec 15, 2022
7c77ae1
Merge branch 'dev-1.x' into dcff_deploy_revision
spynccat Dec 16, 2022
a957099
update fix_subnet with dev-1.x
spynccat Dec 16, 2022
9821668
update comments
spynccat Dec 16, 2022
e4c683b
update docs
spynccat Dec 16, 2022
afb68a0
update registry
spynccat Dec 16, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
12 changes: 10 additions & 2 deletions configs/pruning/mmcls/dcff/dcff_compact_resnet_8xb32_in1k.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -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'))
7 changes: 5 additions & 2 deletions configs/pruning/mmcls/dcff/dcff_resnet_8xb32_in1k.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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')
141 changes: 141 additions & 0 deletions configs/pruning/mmcls/dcff/fix_subnet.json
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -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"
}
}
Loading