mmrazor/tests/data/test_registry/subnet.json

141 lines
4.3 KiB
JSON

{
"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"
}
}