Skip to content

Commit

Permalink
Pruning supported tweaks (#1256)
Browse files Browse the repository at this point in the history
* [CVS-90400, CVS-91015] NNCF pruning supported tweaks (#1248)

* Turned off pruning_support visibility for anomaly models (CVS-91015)

* Disabled pruning for EfficientNet-V2-S (CVS-90400)

* Removed filter pruning definition from compression config; tweaked visible_in_ui for all anomaly tasks
  • Loading branch information
nikita-savelyevv authored Sep 15, 2022
1 parent 956bd3f commit 94b3773
Show file tree
Hide file tree
Showing 11 changed files with 13 additions and 20 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -190,7 +190,8 @@ nncf_optimization:
operator: AND
rules: []
type: UI_RULES
visible_in_ui: true
value: false
visible_in_ui: false
warning: null
type: PARAMETER_GROUP
visible_in_ui: true
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -130,7 +130,7 @@ nncf_optimization:
rules: []
type: UI_RULES
value: false
visible_in_ui: true
visible_in_ui: false
warning: null
type: PARAMETER_GROUP
visible_in_ui: true
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -259,7 +259,7 @@ nncf_optimization:
rules: []
type: UI_RULES
value: false
visible_in_ui: true
visible_in_ui: false
warning: null
type: PARAMETER_GROUP
visible_in_ui: true
Expand Down
3 changes: 2 additions & 1 deletion external/anomaly/configs/detection/draem/configuration.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -190,7 +190,8 @@ nncf_optimization:
operator: AND
rules: []
type: UI_RULES
visible_in_ui: true
value: false
visible_in_ui: false
warning: null
type: PARAMETER_GROUP
visible_in_ui: true
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -130,7 +130,7 @@ nncf_optimization:
rules: []
type: UI_RULES
value: false
visible_in_ui: true
visible_in_ui: false
warning: null
type: PARAMETER_GROUP
visible_in_ui: true
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -259,7 +259,7 @@ nncf_optimization:
rules: []
type: UI_RULES
value: false
visible_in_ui: true
visible_in_ui: false
warning: null
type: PARAMETER_GROUP
visible_in_ui: true
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -190,7 +190,8 @@ nncf_optimization:
operator: AND
rules: []
type: UI_RULES
visible_in_ui: true
value: false
visible_in_ui: false
warning: null
type: PARAMETER_GROUP
visible_in_ui: true
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -130,7 +130,7 @@ nncf_optimization:
rules: []
type: UI_RULES
value: false
visible_in_ui: true
visible_in_ui: false
warning: null
type: PARAMETER_GROUP
visible_in_ui: true
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -259,7 +259,7 @@ nncf_optimization:
rules: []
type: UI_RULES
value: false
visible_in_ui: true
visible_in_ui: false
warning: null
type: PARAMETER_GROUP
visible_in_ui: true
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -44,16 +44,6 @@
},
"nncf_config": {
"compression": [
{
"algorithm": "filter_pruning",
"pruning_init": 0.1,
"params": {
"schedule": "baseline",
"pruning_flops_target": 0.1,
"filter_importance": "geometric_median",
"prune_downsample_convs": true
}
},
{
"algorithm": "quantization",
"preset": "mixed",
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -43,7 +43,7 @@ hyper_parameters:
enable_pruning:
default_value: false
pruning_supported:
default_value: true
default_value: false
maximal_accuracy_degradation:
default_value: 1.0

Expand Down

0 comments on commit 94b3773

Please sign in to comment.