[TF] Custom activation fix (#557)

* fix for #547
openvinotoolkit · Mar 9, 2021 · 722871a · 722871a
1 parent d83b5a0
commit 722871a
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 7 deletions.
diff --git a/beta/nncf/tensorflow/graph/utils.py b/beta/nncf/tensorflow/graph/utils.py
@@ -59,7 +59,7 @@ def get_custom_objects(model):
             custom_objects[layer.__class__.__name__] = layer.__class__
         if layer.__class__.__name__ == 'Activation':
             if layer.activation.__name__ not in keras_activations:
-                custom_objects[layer.activation.__name__] = layer.__class__(layer.activation)
+                custom_objects[layer.activation.__name__] = layer.activation
     return custom_objects
 
 

diff --git a/beta/tests/tensorflow/data/reference_graphs/quantized/asymmetric/per_channel/yolo_v4.pb b/beta/tests/tensorflow/data/reference_graphs/quantized/asymmetric/per_channel/yolo_v4.pb
diff --git a/beta/tests/tensorflow/data/reference_graphs/quantized/symmetric/per_tensor/yolo_v4.pb b/beta/tests/tensorflow/data/reference_graphs/quantized/symmetric/per_tensor/yolo_v4.pb
diff --git a/beta/tests/tensorflow/data/reference_graphs/sparsity/magnitude_sparsity/yolo_v4.pb b/beta/tests/tensorflow/data/reference_graphs/sparsity/magnitude_sparsity/yolo_v4.pb