fix pruning of syncbatchnormalization layer (openvinotoolkit#737)

vshampor · May 26, 2021 · 5a0ce47 · 5a0ce47
1 parent f2cc93d
commit 5a0ce47
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/beta/nncf/tensorflow/layers/common.py b/beta/nncf/tensorflow/layers/common.py
@@ -33,7 +33,8 @@
 LAYERS_WITH_WEIGHTS.update(TF_LAYERS_WITH_WEIGHTS)
 
 SPECIAL_LAYERS_WITH_WEIGHTS = {
-    'BatchNormalization': {WEIGHT_ATTR_NAME: 'gamma', BIAS_ATTR_NAME: 'beta'}
+    'BatchNormalization': {WEIGHT_ATTR_NAME: 'gamma', BIAS_ATTR_NAME: 'beta'},
+    'SyncBatchNormalization': {WEIGHT_ATTR_NAME: 'gamma', BIAS_ATTR_NAME: 'beta'}
 }
 
 ALL_LAYERS_WITH_WEIGHTS = {}

diff --git a/beta/tests/tensorflow/data/reference_graphs/pruning/filter_pruning/yolo_v4.pb b/beta/tests/tensorflow/data/reference_graphs/pruning/filter_pruning/yolo_v4.pb