minor last fixes

sony · Jan 6, 2025 · 8040d69 · 8040d69
1 parent c2016ea
commit 8040d69
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/model_compression_toolkit/core/common/quantization/quantization_config.py b/model_compression_toolkit/core/common/quantization/quantization_config.py
@@ -24,7 +24,7 @@
 
 class CustomOpsetLayers(NamedTuple):
     """
-    This struct is used to define a set of operators to map to a custom operator set defined in the TPC.
+    This struct is used to define a set of operators to be used for setting a custom operator set in the TPC.
 
     Args:
         operators: a list of framework operators to map to a certain custom opset name.

diff --git a/model_compression_toolkit/target_platform_capabilities/schema/v1.py b/model_compression_toolkit/target_platform_capabilities/schema/v1.py
@@ -37,7 +37,7 @@ class OperatorSetNames(str, Enum):
     L2NORM = "L2Norm"
     RELU = "ReLU"
     RELU6 = "ReLU6"
-    LEAKY_RELU = "LEAKYReLU"
+    LEAKY_RELU = "LeakyReLU"
     ELU = "Elu"
     HARD_TANH = "HardTanh"
     ADD = "Add"